custom-diffusion-library
/

cat

stable-diffusion

stable-diffusion-diffusers

Model card Files Files and versions

nupurkmr9 commited on Jan 5, 2023

Commit

60840f9

·

1 Parent(s): fe8c46c

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -29,7 +29,7 @@ model_id = "CompVis/stable-diffusion-v1-4"
 pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
 pipe = pipe.to(device)
-diffuser_training.load_model(pipe.text_encoder, pipe.tokenizer, pipe.unet, 'custom_diffusion_cat.bin')
 prompt = "<new1> cat swimming in a pool"
 images = pipe(prompt, num_inference_steps=200, guidance_scale=6., eta=1.).images

 pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
 pipe = pipe.to(device)
+diffuser_training.load_model(pipe.text_encoder, pipe.tokenizer, pipe.unet, 'cat.bin')
 prompt = "<new1> cat swimming in a pool"
 images = pipe(prompt, num_inference_steps=200, guidance_scale=6., eta=1.).images