custom-diffusion-library
/

cat

stable-diffusion

stable-diffusion-diffusers

Model card Files Files and versions

nupurkmr9 commited on Dec 19, 2022

Commit

f6d8e85

·

1 Parent(s): 4f64a1d

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -27,7 +27,7 @@ from src import diffuser_training
 device = 'cuda'
 pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
-pipe = pipe.to(self.device)
 weight_path = 'custom_diffusion_cat.bin'
 diffuser_training.load_model(pipe.text_encoder, pipe.tokenizer, pipe.unet, weight_path, '<new1>')

 device = 'cuda'
 pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
+pipe = pipe.to(device)
 weight_path = 'custom_diffusion_cat.bin'
 diffuser_training.load_model(pipe.text_encoder, pipe.tokenizer, pipe.unet, weight_path, '<new1>')