openai
/

diffusers-cd_cat256_lpips

ConsistencyModelPipeline

generative model

unconditional image generation

consistency-model

Model card Files Files and versions

patrickvonplaten commited on Jul 5, 2023

Commit

657a65d

·

1 Parent(s): 8de63a0

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -45,7 +45,7 @@ The `diffusers` pipeline for the `cd_cat256_lpips` model can be downloaded as fo
 ```python
 from diffusers import ConsistencyModelPipeline
-pipe = ConsistencyModelPipeline.from_pretrained("dg845/diffusers-cd_cat256_lpips")
 ```
 ## Usage
@@ -61,7 +61,7 @@ from diffusers import ConsistencyModelPipeline
 device = "cuda"
 # Load the cd_cat256_lpips checkpoint.
-model_id_or_path = "dg845/diffusers-cd_cat256_lpips"
 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_cat256_lpips")
 ```
 ## Usage
 device = "cuda"
 # Load the cd_cat256_lpips checkpoint.
+model_id_or_path = "openai/diffusers-cd_cat256_lpips"
 pipe = ConsistencyModelPipeline.from_pretrained(model_id_or_path, torch_dtype=torch.float16)
 pipe.to(device)