hunyuandit

README.md

@@ -33,6 +33,8 @@
 
     pip install flash_attn-2.0.4+das1.0+82379d7.abi0.dtk2404.torch2.1-cp310-cp310-manylinux2014_x86_64.whl  （开发者社区）
 
+    pip install bitsandbytes-0.42.0-py3-none-any.whl  （whl文件夹中）
+
 
 ### Dockerfile（方法二）
 
@@ -46,6 +48,8 @@
 
     pip install flash_attn-2.0.4+das1.0+82379d7.abi0.dtk2404.torch2.1-cp310-cp310-manylinux2014_x86_64.whl  （开发者社区）
 
+    pip install bitsandbytes-0.42.0-py3-none-any.whl  （whl文件夹中）
+
 
 ### Anaconda (方法三)
 1、关于本项目DCU显卡所需的特殊深度学习库可从光合开发者社区下载安装：
@@ -66,6 +70,8 @@ Tips：以上dtk驱动、python、torch等DCU相关工具版本需要严格一
 
     pip install timm --no-deps
 
+    pip install bitsandbytes-0.42.0-py3-none-any.whl  （whl文件夹中）
+
 ## 数据集
 
 无
@@ -144,7 +150,6 @@ https://hf-mirror.com/Tencent-Hunyuan/HunyuanDiT/tree/main
     python sample_t2i.py --prompt "飞流直下三千尺疑是银河落九天" --image-size 1280 768
 
     # Prompt Enhancement + Text-to-Image. DialogGen loads with 4-bit quantization, but it may loss performance.
-    # 版本问题，暂不支持
     python sample_t2i.py --prompt "飞流直下三千尺疑是银河落九天"  --load-4bit