Spaces:

SNA-AI
/

ImageGenerator

Sleeping

SNA-AI commited on Apr 17

Commit

8b3da90

verified ·

1 Parent(s): 99bea0c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,7 +2,7 @@ import torch
 import gradio as gr
 from diffusers import StableDiffusionPipeline
-model_id = "runwayml/stable-diffusion-v1-5"
 device = "cuda" if torch.cuda.is_available() else "cpu"
 pipe = StableDiffusionPipeline.from_pretrained(

 import gradio as gr
 from diffusers import StableDiffusionPipeline
+model_id = "stabilityai/sd-turbo"
 device = "cuda" if torch.cuda.is_available() else "cpu"
 pipe = StableDiffusionPipeline.from_pretrained(