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")
|