eurecom-ds
/

scoresdeve-ema-multi-dsprites-64

Unconditional Image Generation

Model card Files Files and versions Community

giulio98 commited on Apr 1, 2024

Commit

e3c0259

·

verified ·

1 Parent(s): 64858ca

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -13,7 +13,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model_id = "eurecom-ds/scoresdeve-ema-multi-dsprites-64"
 # load model and scheduler
-pipe = DiffusionPipeline.from_pretrained(model_idm, trust_remote_code=True)
 pipe.to(device)

 model_id = "eurecom-ds/scoresdeve-ema-multi-dsprites-64"
 # load model and scheduler
+pipe = DiffusionPipeline.from_pretrained(model_id, trust_remote_code=True)
 pipe.to(device)