soiz1 commited on
Commit
f51b0d0
Β·
verified Β·
1 Parent(s): 52f2486

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -156,6 +156,7 @@ hf_token = os.getenv("token")
156
  model_path = "/home/user/stable-diffusion-webui/models/Stable-diffusion/flux1-dev.safetensors"
157
  if os.path.exists(model_path):
158
  os.remove(model_path)
 
159
 
160
  # wget でフゑむルをダウンロード
161
  cmd = f'wget --header="Authorization: Bearer {hf_token}" -O {model_path} "https://huggingface.co/black-forest-labs/FLUX.1-dev/resolve/main/flux1-dev.safetensors"'
 
156
  model_path = "/home/user/stable-diffusion-webui/models/Stable-diffusion/flux1-dev.safetensors"
157
  if os.path.exists(model_path):
158
  os.remove(model_path)
159
+ os.system("set COMMANDLINE_ARGS=--skip-torch-cuda-test --no-half --use-cpu all")
160
 
161
  # wget でフゑむルをダウンロード
162
  cmd = f'wget --header="Authorization: Bearer {hf_token}" -O {model_path} "https://huggingface.co/black-forest-labs/FLUX.1-dev/resolve/main/flux1-dev.safetensors"'