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
|