Spaces:

ahmed-eisa
/

genai_service

Running

ahmed-eisa commited on 17 days ago

Commit

be3cdfa

1 Parent(s): 5b43cd0

fixed loading audio model

Files changed (1) hide show

models.py CHANGED Viewed

@@ -16,7 +16,8 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 def load_audio_model() -> tuple[BarkProcessor, BarkModel]:
     processor = AutoProcessor.from_pretrained("suno/bark-small", device=device)
-    model = AutoModel.from_pretrained("suno/bark-small", device=device)
     return processor, model
 def generate_audio(

 def load_audio_model() -> tuple[BarkProcessor, BarkModel]:
     processor = AutoProcessor.from_pretrained("suno/bark-small", device=device)
+    model = AutoModel.from_pretrained("suno/bark-small")
+    model.to(device)
     return processor, model
 def generate_audio(