dg845 commited on
Commit
dd6eaad
1 Parent(s): 9f8c855

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -29,7 +29,7 @@ The authors call this procedure "consistency distillation (CD)".
29
  Consistency models can also be trained from scratch to generate clean images from a noisy image and timestep, which the authors call "consistency training (CT)".
30
 
31
  This model is a `diffusers`-compatible version of the [cd_imagenet64_l2.pt](https://github.com/openai/consistency_models#pre-trained-models) checkpont from the [original code and model release](https://github.com/openai/consistency_models).
32
- This model was distilled from an [EDM model](https://arxiv.org/pdf/2206.00364.pdf) trained on the ImageNet 64x64 dataset, using the [L2 distance](https://en.wikipedia.org/wiki/Norm_(mathematics)#Euclidean_norm) as the measure of closeness.
33
  See the [original model card](https://github.com/openai/consistency_models/blob/main/model-card.md) for more information.
34
 
35
  ## Download
@@ -63,7 +63,7 @@ pipe.to(device)
63
 
64
  # Onestep Sampling
65
  image = pipe(num_inference_steps=1).images[0]
66
- image.save("consistency_model_onestep_sample.png")
67
 
68
  # Onestep sampling, class-conditional image generation
69
  # ImageNet-64 class label 145 corresponds to king penguins
 
29
  Consistency models can also be trained from scratch to generate clean images from a noisy image and timestep, which the authors call "consistency training (CT)".
30
 
31
  This model is a `diffusers`-compatible version of the [cd_imagenet64_l2.pt](https://github.com/openai/consistency_models#pre-trained-models) checkpont from the [original code and model release](https://github.com/openai/consistency_models).
32
+ This model was distilled (via consistency distillation (CD)) from an [EDM model](https://arxiv.org/pdf/2206.00364.pdf) trained on the ImageNet 64x64 dataset, using the [L2 distance](https://en.wikipedia.org/wiki/Norm_(mathematics)#Euclidean_norm) as the measure of closeness.
33
  See the [original model card](https://github.com/openai/consistency_models/blob/main/model-card.md) for more information.
34
 
35
  ## Download
 
63
 
64
  # Onestep Sampling
65
  image = pipe(num_inference_steps=1).images[0]
66
+ image.save("cd_imagenet64_l2_onestep_sample.png")
67
 
68
  # Onestep sampling, class-conditional image generation
69
  # ImageNet-64 class label 145 corresponds to king penguins