webui_test / app.py
John6666's picture
Upload 321 files
f29a5f4 verified
import os
if os.environ.get("SPACES_ZERO_GPU") is not None:
import spaces
else:
class spaces:
@staticmethod
def GPU(func):
def wrapper(*args, **kwargs):
return func(*args, **kwargs)
return wrapper
import sys
import subprocess
import shutil
from pathlib import Path
from utils import set_token, git_clone, run, get_file
HF_TOKEN = os.environ.get("HF_TOKEN")
set_token(HF_TOKEN)
@spaces.GPU
def fake_gpu():
pass
BASE_DIR = str(Path(__file__).resolve().parent.resolve())
WEBUI_DIR = str(Path(BASE_DIR, "webui").resolve())
#os.environ["PYTHONPATH"] = os.environ.get("PYTHONPATH") + f":{WEBUI_DIR}:."
#git_clone("https://github.com/AUTOMATIC1111/stable-diffusion-webui.git", BASE_DIR)
os.chdir(WEBUI_DIR)
#run(f"python3 launch.py --precision full --no-half --no-half-vae --enable-insecure-extension-access --medvram --always-cpu --skip-torch-cuda-test --disable-gpu-warning --enable-console-prompts")
run(f"python3 launch.py --precision full --no-half --no-half-vae --enable-insecure-extension-access --medvram --skip-torch-cuda-test --enable-console-prompts")