IT2091024v2

Paused

Pijush2023 commited on Jun 27, 2024

Commit

a5dd543

verified ·

1 Parent(s): 1414013

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1232,7 +1232,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):