takuoko
/

small-sd-dreambooth-lora-dog

Model card Files Files and versions Community

takuoko commited on Oct 24, 2023

Commit

7b31279

•

1 Parent(s): 1cfea74

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -23,7 +23,7 @@ prompt = 'A photo of sks dog in a bucket'
 pipe = DiffusionPipeline.from_pretrained(
     'segmind/small-sd', torch_dtype=torch.float16)
 pipe.to('cuda')
-pipe.load_lora_weights(checkpoint)
 image = pipe(
     prompt,

 pipe = DiffusionPipeline.from_pretrained(
     'segmind/small-sd', torch_dtype=torch.float16)
 pipe.to('cuda')
+pipe.load_lora_weights(checkpoint, weight_name='pytorch_lora_weights.bin')
 image = pipe(
     prompt,