akameswa commited on
Commit
b89fe5f
1 Parent(s): 60458f9

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +11 -0
README.md CHANGED
@@ -20,4 +20,15 @@ PeftModel.from_pretrained(pipe.unet, adapter_id)
20
  prompt = "a dog wearing a knitted hat on the floor"
21
  image = pipe(prompt=prompt, num_inference_steps=4, guidance_scale=1.0).images[0]
22
  display(image)
 
 
 
 
 
 
 
 
 
 
 
23
  ```
 
20
  prompt = "a dog wearing a knitted hat on the floor"
21
  image = pipe(prompt=prompt, num_inference_steps=4, guidance_scale=1.0).images[0]
22
  display(image)
23
+ ```
24
+
25
+ # Saving complete model
26
+ ```python
27
+ pipe.fuse_lora(lora_scale=1.0)
28
+ pipe.unload_lora_weights()
29
+
30
+ for param in pipe.unet.parameters():
31
+ param.data = param.data.contiguous()
32
+
33
+ pipe.save_pretrained("./lcm-tiny-sd")
34
  ```