Spaces:

OpenMOSS-Team
/

MOSS-Speech

Running on Zero

singularitys0 commited on Sep 30

Commit

cc0a32f

1 Parent(s): 107b307

fix bug

Files changed (2) hide show

utils/__pycache__/interface.cpython-310.pyc CHANGED Viewed

Binary files a/utils/__pycache__/interface.cpython-310.pyc and b/utils/__pycache__/interface.cpython-310.pyc differ

utils/interface.py CHANGED Viewed

@@ -27,7 +27,7 @@ class Inference:
         self.device = device
         self.processor = AutoProcessor.from_pretrained(
             model_path,
-            codec_path=codec_path if codec_path else "fnlp/MOSS-Speech-Codec",
             device=self.device,
             trust_remote_code=True,
         )
@@ -67,7 +67,7 @@ class Inference:
         if len(output_modalities) != 1:
             raise ValueError("Expected exactly one output modality based on task.")
-        inputs = self.processor(full_conversation, output_modalities)
         stopping_criteria = [
             MIMOStopper(self.processor.tokenizer.pad_token_id),

         self.device = device
         self.processor = AutoProcessor.from_pretrained(
             model_path,
+            codec_path=codec_path if codec_path else "fnlp/MOSS-Speech",
             device=self.device,
             trust_remote_code=True,
         )
         if len(output_modalities) != 1:
             raise ValueError("Expected exactly one output modality based on task.")
+        inputs = self.processor([full_conversation], output_modalities)
         stopping_criteria = [
             MIMOStopper(self.processor.tokenizer.pad_token_id),