IT2091024v2

Paused

Pijush2023 commited on Jun 28, 2024

Commit

e8f01c7

verified ·

1 Parent(s): ff4ddb2

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1304,7 +1304,7 @@ def generate_audio_elevenlabs(text):
         return None
 # Stable Diffusion setup
-pipe = StableDiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-3-base", torch_dtype=torch.float16)
 pipe = pipe.to("cuda")
 def generate_image(prompt):

         return None
 # Stable Diffusion setup
+pipe = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3-medium-diffusers", torch_dtype=torch.float16)
 pipe = pipe.to("cuda")
 def generate_image(prompt):