Update README.md
Browse files
README.md
CHANGED
@@ -100,7 +100,7 @@ import diffusers
|
|
100 |
device = "cuda"
|
101 |
dtype = torch.float16
|
102 |
model_path = "Disty0/sotediffusion-v2"
|
103 |
-
pipe = diffusers.AutoPipelineForText2Image.from_pretrained(model_path,
|
104 |
|
105 |
# de-dupe
|
106 |
pipe.decoder_pipe.text_encoder = pipe.text_encoder = None # nothing uses this
|
|
|
100 |
device = "cuda"
|
101 |
dtype = torch.float16
|
102 |
model_path = "Disty0/sotediffusion-v2"
|
103 |
+
pipe = diffusers.AutoPipelineForText2Image.from_pretrained(model_path, torch_dtype=dtype)
|
104 |
|
105 |
# de-dupe
|
106 |
pipe.decoder_pipe.text_encoder = pipe.text_encoder = None # nothing uses this
|