xy1993 commited on
Commit
12443d5
·
verified ·
1 Parent(s): 79324fe

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -3
README.md CHANGED
@@ -92,14 +92,17 @@ TensorArt Stable Diffusion 3.5 Medium Turbo (SD3.5M Turbo) is a high-performance
92
  from diffusers import StableDiffusion3Pipeline
93
  import numpy as np
94
  from safetensors.torch import load_file
 
 
 
 
95
 
96
  pipe = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3.5-medium", torch_dtype=torch.float16,)
97
 
98
  pipe = pipe.to("cuda")
99
 
100
- lora_weights_dir = 'tensorart/stable-diffusion-3.5-medium-turbo/lora_sd3.5m_turbo_8steps.safetensors'
101
- pcm_lora_weight = load_file(lora_weights_dir)
102
-
103
 
104
 
105
  pipe = pipe.to("cuda")
 
92
  from diffusers import StableDiffusion3Pipeline
93
  import numpy as np
94
  from safetensors.torch import load_file
95
+ from huggingface_hub import hf_hub_download
96
+
97
+ repo = "tensorart/stable-diffusion-3.5-medium-turbo"
98
+ ckpt = "lora_sd3.5m_turbo_8steps.safetensors"
99
 
100
  pipe = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3.5-medium", torch_dtype=torch.float16,)
101
 
102
  pipe = pipe.to("cuda")
103
 
104
+ pipe.load_lora_weights(hf_hub_download(repo, ckpt))
105
+ pipe.fuse_lora()
 
106
 
107
 
108
  pipe = pipe.to("cuda")