Spaces:

jayparmr
/

DreamShaper

Running

jayparmr commited on May 1, 2023

Commit

15fa9bb

•

1 Parent(s): 49c5279

Update

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,12 +1,11 @@
 import gradio as gr
 import torch
-# from diffusers import DiffusionPipeline
-from diffusers import StableDiffusionPipeline
-SAFETENSORS_PATH = "./dreamshaper_5BakedVae.safetensors"
 def generate(prompt):
-    pipeline = StableDiffusionPipeline.from_pretrained(SAFETENSORS_PATH , torch_dtype = torch.float16 , use_safetensors = True)
     pipeline.to("cuda")
     return pipeline(prompt).images[0]

 import gradio as gr
 import torch
+from diffusers import DiffusionPipeline
+# from diffusers import StableDiffusionPipeline
 def generate(prompt):
+    pipeline = DiffusionPipeline.from_pretrained("Lykon/DreamShaper")
     pipeline.to("cuda")
     return pipeline(prompt).images[0]