MindVR commited on
Commit
bc2c49d
·
verified ·
1 Parent(s): 05a9b7e

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -29
requirements.txt CHANGED
@@ -1,29 +1,3 @@
1
- # Install torch first with the correct CUDA version
2
- # Using cu121 index as specified. Base image is cu123, which is compatible.
3
- --index-url https://download.pytorch.org/whl/cu121
4
- torch==2.2.2
5
- torchvision==0.17.2
6
- torchaudio==2.2.2
7
-
8
- # Install dependencies compatible with Unsloth and PyTorch
9
- xformers==0.0.25
10
- bitsandbytes==0.43.1
11
- accelerate==0.29.3
12
- peft==0.10.0
13
- trl==0.17.0
14
- transformers==4.40.2
15
-
16
- # Install unsloth from the main branch (less stable than a specific version)
17
- # This attempts to bypass issues fetching specific historical commits.
18
- unsloth[cu121] @ git+https://github.com/unslothai/unsloth.git@main
19
-
20
- # Other packages
21
- sentencepiece
22
- protobuf
23
- datasets
24
- huggingface_hub>=0.19
25
- hf_xet>=1.0.0,<2.0.0
26
- hf-transfer>=0.1.4
27
- click<8.1
28
- pydantic~=1.0
29
- gradio>=4.0.0
 
1
+ transformers
2
+ torch
3
+ gradio