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