mountdrive / app.py
raylander's picture
Upload app.py
586dcfc
raw
history blame
1.01 kB
import os
from subprocess import getoutput
gpu_info = getoutput('nvidia-smi')
if("A10G" in gpu_info):
os.system(f"pip install -q https://github.com/camenduru/stable-diffusion-webui-colab/releases/download/0.0.16/xformers-0.0.16+814314d.d20230118-cp38-cp38-linux_x86_64.whl")
elif("T4" in gpu_info):
os.system(f"pip install -q https://github.com/camenduru/stable-diffusion-webui-colab/releases/download/0.0.16/xformers-0.0.16+814314d.d20230118-cp38-cp38-linux_x86_64.whl")
# os.system(f"git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui.git /home/user/app/stable-diffusion-webui")
# os.chdir("/home/user/app/stable-diffusion-webui")
os.system(f"conda create -n dsd python=3.10 -y")
os.system(f"conda activate dsd")
os.system(f"git clone https://github.com/deforum-art/deforum-stable-diffusion.git")
os.system(f"cd deforum-stable-diffusion")
os.system(f"python deforum-stable-diffusion/install_requirements.py")
os.system(f"python deforum-stable-diffusion/Deforum_Stable_Diffusion.py")