Hugging Face's logo

openai
/
diffusers-cd_cat256_lpips

Diffusers
ConsistencyModelPipeline
generative model
unconditional image generation
consistency-model
Model card Files Community
patrickvonplaten commited on
Commit
657a65d
·
1 Parent(s): 8de63a0

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -45,7 +45,7 @@ The `diffusers` pipeline for the `cd_cat256_lpips` model can be downloaded as fo
45
  ```python
46
  from diffusers import ConsistencyModelPipeline
47
 
48
- pipe = ConsistencyModelPipeline.from_pretrained("dg845/diffusers-cd_cat256_lpips")
49
  ```
50
 
51
  ## Usage
@@ -61,7 +61,7 @@ from diffusers import ConsistencyModelPipeline
61
 
62
  device = "cuda"
63
  # Load the cd_cat256_lpips checkpoint.
64
- model_id_or_path = "dg845/diffusers-cd_cat256_lpips"
65
  pipe = ConsistencyModelPipeline.from_pretrained(model_id_or_path, torch_dtype=torch.float16)
66
  pipe.to(device)
67
 
 
45
  ```python
46
  from diffusers import ConsistencyModelPipeline
47
 
48
+ pipe = ConsistencyModelPipeline.from_pretrained("openai/diffusers-cd_cat256_lpips")
49
  ```
50
 
51
  ## Usage
 
61
 
62
  device = "cuda"
63
  # Load the cd_cat256_lpips checkpoint.
64
+ model_id_or_path = "openai/diffusers-cd_cat256_lpips"
65
  pipe = ConsistencyModelPipeline.from_pretrained(model_id_or_path, torch_dtype=torch.float16)
66
  pipe.to(device)
67