Upload 2 files

app.py

@@ -10,8 +10,10 @@ def current_time():
 
 print(f"[{current_time()}] 开始部署空间...")
 
+"""
 print(f"[{current_time()}] 日志：安装 - 必要包")
 os.system("pip install -r ./requirements.txt")
+"""
 print(f"[{current_time()}] 日志：安装 - gsutil")
 os.system("pip install gsutil")
 print(f"[{current_time()}] 日志：Git - 克隆 Github 的 T5X 训练框架到当前目录")
@@ -24,7 +26,12 @@ print(f"[{current_time()}] 日志：Python - 使用 pip 安装 当前目录内
 os.system("python3 -m pip install -e .")
 print(f"[{current_time()}] 日志：Python - 更新 Python 包管理器 pip")
 os.system("python3 -m pip install --upgrade pip")
+print(f"[{current_time()}] 日志：安装 - langchain")
+os.system("pip install langchain")
+print(f"[{current_time()}] 日志：安装 - sentence-transformers")
+os.system("pip install sentence-transformers")
 
+"""
 # 安装 airio
 print(f"[{current_time()}] 日志：Git - 克隆 Github 的 airio 到当前目录")
 os.system("git clone --branch=main https://github.com/google/airio")
@@ -32,12 +39,15 @@ print(f"[{current_time()}] 日志：文件 - 移动 airio 到当前目录并重
 os.system("mv airio airio_tmp; mv airio_tmp/* .; rm -r airio_tmp")
 print(f"[{current_time()}] 日志：Python - 使用 pip 安装 当前目录内的 Python 包")
 os.system("python3 -m pip install -e .")
+"""
 
 # 安装 mt3
 print(f"[{current_time()}] 日志：Git - 克隆 Github 的 MT3 模型到当前目录")
 os.system("git clone --branch=main https://github.com/magenta/mt3")
 print(f"[{current_time()}] 日志：文件 - 移动 mt3 到当前目录并重命名为 mt3_tmp 并删除")
 os.system("mv mt3 mt3_tmp; mv mt3_tmp/* .; rm -r mt3_tmp")
+print(f"[{current_time()}] 日志：Python - 使用 pip 从 storage.googleapis.com 安装 jax[cuda11_local] nest-asyncio pyfluidsynth")
+os.system("python3 -m pip install jax[cuda11_local] nest-asyncio pyfluidsynth==1.3.0 -e . -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html")
 print(f"[{current_time()}] 日志：Python - 使用 pip 安装 当前目录内的 Python 包")
 os.system("python3 -m pip install -e .")
 print(f"[{current_time()}] 日志：安装 - TensorFlow CPU")