Spaces:

Walmart-the-bag
/

Juggernaut-X-v10

Running on Zero

Walmart-the-bag commited on Apr 24

Commit

c4cc804

•

1 Parent(s): 3075cc3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,11 +1,11 @@
-from diffusers import StableDiffusionPipeline
 import torch
 from gradio import Interface, Image
 import gradio as gr
 import spaces
 model_id = "RunDiffusion/Juggernaut-X-v10"
-pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
 pipe = pipe.to("cuda")
 @spaces.GPU()

+from diffusers import StableDiffusionXLPipeline
 import torch
 from gradio import Interface, Image
 import gradio as gr
 import spaces
 model_id = "RunDiffusion/Juggernaut-X-v10"
+pipe = StableDiffusionXLPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
 pipe = pipe.to("cuda")
 @spaces.GPU()