Update README.md
Browse files
README.md
CHANGED
@@ -93,7 +93,7 @@ TensorArt Stable Diffusion 3.5 Medium Turbo (SD3.5M Turbo) is a high-performance
|
|
93 |
import numpy as np
|
94 |
from safetensors.torch import load_file
|
95 |
|
96 |
-
pipe = StableDiffusion3Pipeline.from_pretrained("
|
97 |
|
98 |
pipe = pipe.to("cuda")
|
99 |
|
|
|
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 |
|