start3406 commited on
Commit
0ad3783
·
verified ·
1 Parent(s): a63d56e

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +8 -10
requirements.txt CHANGED
@@ -1,11 +1,9 @@
1
  gradio
2
- torch # 根据你的环境可能需要指定版本,如 torch torchvision torchaudio
3
- transformers>=4.24.0 # 需要较新版本支持pipeline和模型
4
- diffusers>=0.11.1 # 需要较新版本支持Stable Diffusion Pipeline
5
- accelerate # 推荐,用于优化模型加载和运行,特别是GPU offload
6
- sentencepiece # transformers可能需要
7
- # 如果使用GPU,可能需要安装对应的cuda版本,如:
8
- # pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
9
- # 具体依赖项可能根据你的Python版本和操作系统有所不同
10
- # whisper 需要 ffmpeg 支持,Hugging Face Spaces 基础镜像通常自带
11
- # pip install ffmpeg-python # 或者在系统层面安装 ffmpeg
 
1
  gradio
2
+ torch --index-url https://download.pytorch.org/whl/cpu # Force CPU version of torch
3
+ transformers>=4.24.0
4
+ diffusers>=0.11.1
5
+ openai>=1.0 # Add openai library
6
+ sentencepiece
7
+ # accelerate is likely not needed for pure CPU operation
8
+ # ffmpeg should be available in HF Spaces base images, if not:
9
+ # ffmpeg-python