Update README
Browse files- README.md +1 -1
- README_CN.md +1 -1
README.md
CHANGED
@@ -221,7 +221,7 @@ import torch
|
|
221 |
from hyimage.diffusion.pipelines.hunyuanimage_pipeline import HunyuanImagePipeline
|
222 |
|
223 |
# Supported model_name: hunyuanimage-v2.1, hunyuanimage-v2.1-distilled
|
224 |
-
model_name = "hunyuanimage-v2.1
|
225 |
pipe = HunyuanImagePipeline.from_pretrained(model_name=model_name, torch_dtype='bf16')
|
226 |
pipe = pipe.to("cuda")
|
227 |
|
|
|
221 |
from hyimage.diffusion.pipelines.hunyuanimage_pipeline import HunyuanImagePipeline
|
222 |
|
223 |
# Supported model_name: hunyuanimage-v2.1, hunyuanimage-v2.1-distilled
|
224 |
+
model_name = "hunyuanimage-v2.1"
|
225 |
pipe = HunyuanImagePipeline.from_pretrained(model_name=model_name, torch_dtype='bf16')
|
226 |
pipe = pipe.to("cuda")
|
227 |
|
README_CN.md
CHANGED
@@ -193,7 +193,7 @@ import torch
|
|
193 |
from hyimage.diffusion.pipelines.hunyuanimage_pipeline import HunyuanImagePipeline
|
194 |
|
195 |
# 支持的 model_name:hunyuanimage-v2.1, hunyuanimage-v2.1-distilled
|
196 |
-
model_name = "hunyuanimage-v2.1
|
197 |
pipe = HunyuanImagePipeline.from_pretrained(model_name=model_name, torch_dtype='bf16')
|
198 |
pipe = pipe.to("cuda")
|
199 |
|
|
|
193 |
from hyimage.diffusion.pipelines.hunyuanimage_pipeline import HunyuanImagePipeline
|
194 |
|
195 |
# 支持的 model_name:hunyuanimage-v2.1, hunyuanimage-v2.1-distilled
|
196 |
+
model_name = "hunyuanimage-v2.1"
|
197 |
pipe = HunyuanImagePipeline.from_pretrained(model_name=model_name, torch_dtype='bf16')
|
198 |
pipe = pipe.to("cuda")
|
199 |
|