napstablook911 commited on
Commit
ec1031e
·
verified ·
1 Parent(s): 8e93ddc

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +1 -1
src/streamlit_app.py CHANGED
@@ -44,7 +44,7 @@ def load_stable_audio_model():
44
  try:
45
  # Changed model to stabilityai/stable-audio-open-small
46
  audio_pipeline = StableAudioPipeline.from_pretrained(
47
- "stabilityai/stable-audio-open-small", # <--- MODEL CHANGED HERE
48
  torch_dtype=TORCH_DTYPE
49
  ).to(DEVICE)
50
  return audio_pipeline
 
44
  try:
45
  # Changed model to stabilityai/stable-audio-open-small
46
  audio_pipeline = StableAudioPipeline.from_pretrained(
47
+ "stabilityai/stable-audio-open-1.0",
48
  torch_dtype=TORCH_DTYPE
49
  ).to(DEVICE)
50
  return audio_pipeline