Spaces:
Running
Running
Update src/streamlit_app.py
Browse files- 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-
|
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
|