Luo-Yihong
/

yoso_sd1.5_lora

Model card Files Files and versions Community

Luo-Yihong commited on Apr 8

Commit

b25d8b2

•

1 Parent(s): 7422ed6

Update README.md

Files changed (1) hide show

README.md +2 -1

README.md CHANGED Viewed

@@ -29,8 +29,9 @@ steps = 1
 bs = 1
 latents = ... # maybe some latent codes of real images or SD generation
 latent_mean = latent.mean(dim=0)
 noise = torch.randn([1,bs,64,64])
-input_latent = pipeline.scheduler.add_noise(latent_mean.repeat(bs,1,1,1),noise,T)
 imgs= pipeline(prompt="A photo of a dog",
                     num_inference_steps=steps,
                     num_images_per_prompt = 1,

 bs = 1
 latents = ... # maybe some latent codes of real images or SD generation
 latent_mean = latent.mean(dim=0)
+init_latent = latent_mean.repeat(bs,1,1,1) + latents.std()*torch.randn_like(latents)
 noise = torch.randn([1,bs,64,64])
+input_latent = pipeline.scheduler.add_noise(init_latent,noise,T)
 imgs= pipeline(prompt="A photo of a dog",
                     num_inference_steps=steps,
                     num_images_per_prompt = 1,