Spaces:

lakshmikarpolam
/

LLCM_Medical_Images_Generation

Runtime error

lakshmikarpolam commited on Mar 11

Commit

d4bf7ca

•

1 Parent(s): 5b4f0ac

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ def generate_and_display_image(prompt):
     # Set the loaded scheduler in the pipeline
     pipeline.scheduler = LCMScheduler.from_config(pipeline.scheduler.config)
-    # pipeline.to("cuda")
     # Set the number of inference steps
     inference_steps = 4

     # Set the loaded scheduler in the pipeline
     pipeline.scheduler = LCMScheduler.from_config(pipeline.scheduler.config)
+    pipeline.to("cuda")
     # Set the number of inference steps
     inference_steps = 4