openai
/

diffusers-cd_imagenet64_l2

ConsistencyModelPipeline

generative model

unconditional image generation

consistency-model

Model card Files Files and versions Community

patrickvonplaten commited on Jul 5, 2023

Commit

04420ad

•

1 Parent(s): c44bb87

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -46,7 +46,7 @@ The `diffusers` pipeline for the `cd-imagenet64-l2` model can be downloaded as f
 ```python
 from diffusers import ConsistencyModelPipeline
-pipe = ConsistencyModelPipeline.from_pretrained("openai/consistency-model-pipelines")
 ```
 ## Usage
@@ -62,7 +62,7 @@ from diffusers import ConsistencyModelPipeline
 device = "cuda"
 # Load the cd_imagenet64_l2 checkpoint.
-model_id_or_path = "openai/consistency-model-pipelines"
 pipe = ConsistencyModelPipeline.from_pretrained(model_id_or_path, torch_dtype=torch.float16)
 pipe.to(device)

 ```python
 from diffusers import ConsistencyModelPipeline
+pipe = ConsistencyModelPipeline.from_pretrained("openai/diffusers-cd_imagenet64_l2")
 ```
 ## Usage
 device = "cuda"
 # Load the cd_imagenet64_l2 checkpoint.
+model_id_or_path = "openai/diffusers-cd_imagenet64_l2"
 pipe = ConsistencyModelPipeline.from_pretrained(model_id_or_path, torch_dtype=torch.float16)
 pipe.to(device)