Update README.md
Browse files
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 |
-
|
101 |
-
|
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")
|