KevinNg99 commited on
Commit
d9c9398
·
1 Parent(s): 477b0fd

Update README

Browse files
Files changed (2) hide show
  1. README.md +1 -1
  2. 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-distilled"
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-distilled"
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