Update app.py

app.py

@@ -9,7 +9,8 @@ import torch
 import uuid
 import os
 import soundfile as sf
-from diffusers.pipelines.stable_audio.pipeline_stable_audio import StableAudioPipeline
+from diffusers import DiffusionPipeline
+
 
 
 app = FastAPI()
@@ -38,9 +39,7 @@ def load_pipeline():
         if token:
             login(token=token)
 
-        pipe = StableAudioPipeline.from_pretrained("stabilityai/stable-audio-open-1.0", torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,)
-        pipe.to(device)
-        pipe.enable_attention_slicing()
+        pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-audio-open-1.0",torch_dtype=DTYPE,trust_remote_code=True)
 
         print("Pipeline loaded on", DEVICE)