Hugging Face's logo

Spaces:
LiuZichen
/
MagicQuillV2
Running on Zero

App Files Community
LiuZichen commited on
Commit
1f7b79f
·
verified ·
1 Parent(s): a00240e

Update edit_space.py

Browse files
Files changed (1) hide show
  1. edit_space.py +1 -1
edit_space.py CHANGED
@@ -90,7 +90,7 @@ from utils_node import BlendInpaint, JoinImageWithAlpha, GrowMask, InvertMask, C
90
  TEST_MODE = False
91
 
92
  class KontextEditModel():
93
- def __init__(self, base_model_path="/data0/lzc/FLUX.1-Kontext-dev", device="cuda",
94
  aux_lora_dir="models/v2_ckpt", easycontrol_base_dir="models/v2_ckpt",
95
  aux_lora_weight_name="puzzle_lora.safetensors",
96
  aux_lora_weight=1.0):
 
90
  TEST_MODE = False
91
 
92
  class KontextEditModel():
93
+ def __init__(self, base_model_path="black-forest-labs/FLUX.1-Kontext-dev", device="cuda",
94
  aux_lora_dir="models/v2_ckpt", easycontrol_base_dir="models/v2_ckpt",
95
  aux_lora_weight_name="puzzle_lora.safetensors",
96
  aux_lora_weight=1.0):