Leyla2612 commited on
Commit
ba8ad0d
·
verified ·
1 Parent(s): f4906b9

fix: still conflict ver err

Browse files
Files changed (1) hide show
  1. requirements.txt +28 -12
requirements.txt CHANGED
@@ -1,7 +1,14 @@
1
- torch==2.7.0+cu121
2
- torchvision==0.22.0+cu121
3
- torchaudio==2.7.0+cu121
4
- pytorch-lightning==2.5.1.post0
 
 
 
 
 
 
 
5
  einops==0.8.0
6
  omegaconf==2.3.3
7
  deepspeed==0.16.0
@@ -9,19 +16,28 @@ torchmetrics==1.4.0
9
  webdataset==0.2.6
10
  accelerate==1.0.0
11
  tensorboard==2.17.1
 
 
12
  PyMCubes==0.1.0
13
  trimesh==4.4.9
 
 
 
 
14
  rembg==2.0.59
 
 
 
15
  transformers==4.46.3
16
  diffusers==0.30.3
17
  bitsandbytes==0.44.0
18
- imageio[ffmpeg]==2.35.1
19
- xatlas==0.0.7
20
- plyfile==0.9.4
21
- xformers==0.0.27.post2
22
- --find-links https://download.pytorch.org/whl/cu121
23
- https://download.pytorch.org/whl/cu121/xformers-0.0.28.post1%2Bcu121-cp312-cp312-linux_x86_64.whl # Adjust cho OS
24
- git+https://github.com/NVlabs/nvdiffrast/
25
  huggingface-hub==0.25.2
 
 
26
  onnxruntime-gpu==1.19.2
27
- numpy<3.0 # Tránh conflict transformers
 
 
 
 
 
 
1
+ # === PYTORCH CUDA 12.1 (chính thức hỗ trợ Python 3.11) ===
2
+ --index-url https://download.pytorch.org/whl/cu121
3
+ torch==2.4.1
4
+ torchvision==0.19.1
5
+ torchaudio==2.4.1
6
+
7
+ # === XFORMERS (hỗ trợ cp311 + cu121) ===
8
+ xformers==0.0.27.post2
9
+
10
+ # === ML / Training ===
11
+ pytorch-lightning==2.5.1
12
  einops==0.8.0
13
  omegaconf==2.3.3
14
  deepspeed==0.16.0
 
16
  webdataset==0.2.6
17
  accelerate==1.0.0
18
  tensorboard==2.17.1
19
+
20
+ # === 3D / Mesh ===
21
  PyMCubes==0.1.0
22
  trimesh==4.4.9
23
+ xatlas==0.0.7
24
+ plyfile==0.9.4
25
+
26
+ # === Image / Video ===
27
  rembg==2.0.59
28
+ imageio[ffmpeg]==2.35.1
29
+
30
+ # === HF Ecosystem ===
31
  transformers==4.46.3
32
  diffusers==0.30.3
33
  bitsandbytes==0.44.0
 
 
 
 
 
 
 
34
  huggingface-hub==0.25.2
35
+
36
+ # === ONNX Runtime GPU ===
37
  onnxruntime-gpu==1.19.2
38
+
39
+ # === NVDIFFRAST (git) ===
40
+ git+https://github.com/NVlabs/nvdiffrast.git
41
+
42
+ # === NumPy (tránh conflict) ===
43
+ numpy>=1.23,<2.0