li-yan
/

diffusion-aurora-256

Unconditional Image Generation

Model card Files Files and versions Community

li-yan commited on Jun 20

Commit

4fd02c8

•

1 Parent(s): 94b73c1

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -27,7 +27,7 @@ scheduler = DDIMScheduler.from_pretrained('li-yan/diffusion-aurora-256')
 scheduler.set_timesteps(num_inference_steps=40)
 pipeline = DiffusionPipeline.from_pretrained(
-    'li-yan/diffusion-aurora-256', scheduler=scheduler).to("cuda")
 images = pipeline(num_inference_steps=40).images
 images[0]

 scheduler.set_timesteps(num_inference_steps=40)
 pipeline = DiffusionPipeline.from_pretrained(
+    'li-yan/diffusion-aurora-256', scheduler=scheduler)
 images = pipeline(num_inference_steps=40).images
 images[0]