Yihong Luo commited on
Commit
4dff679
1 Parent(s): 6487dd2

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -14,7 +14,7 @@ bs = 1
14
  latents = ... # maybe some latent codes of real images or SD generation
15
  latent_mean = latent.mean(dim=0)
16
  noise = torch.randn([1,bs,64,64])
17
- input_latent = pipeline.scheduler.add_noise(latent_mean.repeat(bs,1,1,1),noise,1000)
18
  imgs= pipeline(prompt="A photo of a dog",
19
  num_inference_steps=steps,
20
  num_images_per_prompt = 1,
 
14
  latents = ... # maybe some latent codes of real images or SD generation
15
  latent_mean = latent.mean(dim=0)
16
  noise = torch.randn([1,bs,64,64])
17
+ input_latent = pipeline.scheduler.add_noise(latent_mean.repeat(bs,1,1,1),noise,T)
18
  imgs= pipeline(prompt="A photo of a dog",
19
  num_inference_steps=steps,
20
  num_images_per_prompt = 1,