DytaW commited on
Commit
d71995c
·
verified ·
1 Parent(s): 16812a5

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +12 -22
requirements.txt CHANGED
@@ -1,23 +1,13 @@
1
- torch==2.1.0
2
- torchvision==0.16.0
3
- torchaudio==2.1.0
4
- pytorch-lightning==2.1.2
5
- einops
6
- omegaconf
7
- deepspeed
8
- torchmetrics
9
- webdataset
10
- accelerate
11
- tensorboard
12
- PyMCubes
13
- trimesh
14
  rembg
15
- transformers==4.34.1
16
- diffusers==0.19.3
17
- bitsandbytes
18
- imageio[ffmpeg]
19
- xatlas
20
- plyfile
21
- xformers==0.0.22.post7
22
- git+https://github.com/NVlabs/nvdiffrast/
23
- huggingface-hub
 
1
+ # 基礎介面與工具
2
+ gradio==4.44.1
3
+ numpy<2.0.0
4
+ Pillow==10.4.0
5
+
6
+ # 核心 AI 引擎 (補上缺失的這行)
7
+ onnxruntime
 
 
 
 
 
 
8
  rembg
9
+
10
+ # 深度學習框架
11
+ torch
12
+ torchvision
13
+ transformers>=4.40.0