Update README.md
Browse files
README.md
CHANGED
|
@@ -24,6 +24,7 @@ device = 'cuda'
|
|
| 24 |
pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
|
| 25 |
pipe = pipe.to(self.device)
|
| 26 |
|
|
|
|
| 27 |
diffuser_training.load_model(pipe.text_encoder, pipe.tokenizer, pipe.unet, weight_path, '<new1>')
|
| 28 |
prompt = "<new1> cat swimming in a pool"
|
| 29 |
images = pipe(prompt, num_inference_steps=200, guidance_scale=6., eta=1.).images
|
|
|
|
| 24 |
pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
|
| 25 |
pipe = pipe.to(self.device)
|
| 26 |
|
| 27 |
+
weight_path = 'custom_diffusion_cat.bin'
|
| 28 |
diffuser_training.load_model(pipe.text_encoder, pipe.tokenizer, pipe.unet, weight_path, '<new1>')
|
| 29 |
prompt = "<new1> cat swimming in a pool"
|
| 30 |
images = pipe(prompt, num_inference_steps=200, guidance_scale=6., eta=1.).images
|