Spaces:

hsm-kd-master
/

photorealistic-images

Runtime error

fjenett commited on Apr 20, 2023

Commit

4bb0520

•

1 Parent(s): f1c3e3a

adding accelerate

Files changed (2) hide show

app.py CHANGED Viewed

@@ -7,6 +7,8 @@ pipeline = DiffusionPipeline.from_pretrained("dreamlike-art/dreamlike-photoreal-
 # move to GPU if available
 if torch.cuda.is_available():
     pipeline = pipeline.to("cuda")
 def generate(prompts):
   images = pipeline(list(prompts)).images

 # move to GPU if available
 if torch.cuda.is_available():
     pipeline = pipeline.to("cuda")
+else:
+    print('No CUDA, using CPU')
 def generate(prompts):
   images = pipeline(list(prompts)).images

requirements.txt CHANGED Viewed

@@ -1,2 +1,5 @@
 diffusers
-torch

 diffusers
+torch
+gradio
+transformers
+accelerate