Spaces:

pcuenq
/

latent-diffusion-seed

Runtime error

Pedro Cuenca commited on Aug 12, 2022

Commit

2bf201c

•

1 Parent(s): 4b6a41a

Reduce the number of steps while we run on CPU.

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,7 +13,7 @@ from PIL import Image
 SELECT_LABEL = "Select as seed"
 MODEL_ID = "CompVis/ldm-text2im-large-256"
-STEPS = 50
 ETA = 0.3
 GUIDANCE_SCALE = 6
@@ -21,7 +21,6 @@ ldm = DiffusionPipeline.from_pretrained(MODEL_ID)
 import torch
 print(f"cuda: {torch.cuda.is_available()}")
-print(f"device: {torch.cuda.get_device_name()}")
 with gr.Blocks(css=".container { max-width: 800px; margin: auto; }") as demo:
     state = gr.Variable({

 SELECT_LABEL = "Select as seed"
 MODEL_ID = "CompVis/ldm-text2im-large-256"
+STEPS = 25   # while running on CPU
 ETA = 0.3
 GUIDANCE_SCALE = 6
 import torch
 print(f"cuda: {torch.cuda.is_available()}")
 with gr.Blocks(css=".container { max-width: 800px; margin: auto; }") as demo:
     state = gr.Variable({