Update requirements.txt
Browse files- requirements.txt +1 -1
requirements.txt
CHANGED
|
@@ -1,7 +1,7 @@
|
|
| 1 |
fastapi
|
| 2 |
uvicorn[standard] # 標準的な依存関係も含む
|
| 3 |
transformers >= 4.51.3 # PyTorchも一緒に (torchは別途指定も可)
|
| 4 |
-
torch --index-url https://download.pytorch.org/whl/
|
| 5 |
accelerate==1.6.0 # device_map="auto" や大規模モデルのロードに推奨
|
| 6 |
bitsandbytes==0.41.0 # 4bit/8bit量子化に必須 (CUDA環境が必要)
|
| 7 |
sentencepiece # 多くの日本語モデルで必要
|
|
|
|
| 1 |
fastapi
|
| 2 |
uvicorn[standard] # 標準的な依存関係も含む
|
| 3 |
transformers >= 4.51.3 # PyTorchも一緒に (torchは別途指定も可)
|
| 4 |
+
torch --index-url https://download.pytorch.org/whl/cu121 # 特定のCUDAバージョン用
|
| 5 |
accelerate==1.6.0 # device_map="auto" や大規模モデルのロードに推奨
|
| 6 |
bitsandbytes==0.41.0 # 4bit/8bit量子化に必須 (CUDA環境が必要)
|
| 7 |
sentencepiece # 多くの日本語モデルで必要
|