|
@@ -12,7 +12,8 @@ websockets>=12.0
|
|
|
# NVIDIA GPU: pip install torch --index-url https://download.pytorch.org/whl/cu121
|
|
# NVIDIA GPU: pip install torch --index-url https://download.pytorch.org/whl/cu121
|
|
|
# 沐曦 GPU: pip install torch_musa --index-url https://release.mthreads.com/repo/pypi/simple
|
|
# 沐曦 GPU: pip install torch_musa --index-url https://release.mthreads.com/repo/pypi/simple
|
|
|
# 以下为框架和工具库依赖
|
|
# 以下为框架和工具库依赖
|
|
|
-transformers>=4.45.0
|
|
|
|
|
|
|
+# transformers>=4.45.0
|
|
|
|
|
+transformers @ git+https://github.com/huggingface/transformers.git
|
|
|
peft>=0.13.0
|
|
peft>=0.13.0
|
|
|
trl>=0.12.0
|
|
trl>=0.12.0
|
|
|
datasets>=3.0.0
|
|
datasets>=3.0.0
|
