Create requirements.txt
Browse files- requirements.txt +11 -0
requirements.txt
ADDED
|
@@ -0,0 +1,11 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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
|