yasserrmd commited on
Commit
cdec4cd
·
verified ·
1 Parent(s): bdbe728

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ logger = logging.get_logger(__name__)
22
 
23
 
24
  class VibeVoiceDemo:
25
- def __init__(self, model_paths: dict, device: str = "cuda", inference_steps: int = 5):
26
  """
27
  model_paths: dict like {"VibeVoice-1.5B": "microsoft/VibeVoice-1.5B",
28
  "VibeVoice-1.1B": "microsoft/VibeVoice-1.1B"}
 
22
 
23
 
24
  class VibeVoiceDemo:
25
+ def __init__(self, model_paths: dict, device: str = "cuda", inference_steps: int = 5):
26
  """
27
  model_paths: dict like {"VibeVoice-1.5B": "microsoft/VibeVoice-1.5B",
28
  "VibeVoice-1.1B": "microsoft/VibeVoice-1.1B"}