Spaces:
Running
on
Zero
Running
on
Zero
- scripts/process_utils.py +1 -1
scripts/process_utils.py
CHANGED
@@ -47,7 +47,7 @@ def initialize(_use_local=False, use_gpu=False, use_dotenv=False):
|
|
47 |
|
48 |
init_model(use_local)
|
49 |
# model = load_wd14_tagger_model()
|
50 |
-
|
51 |
# refine_gen_pipe = initialize_refine_model()
|
52 |
|
53 |
def load_lora(pipeline, lora_path, adapter_name, alpha=0.75):
|
|
|
47 |
|
48 |
init_model(use_local)
|
49 |
# model = load_wd14_tagger_model()
|
50 |
+
sotai_gen_pipe = initialize_sotai_model()
|
51 |
# refine_gen_pipe = initialize_refine_model()
|
52 |
|
53 |
def load_lora(pipeline, lora_path, adapter_name, alpha=0.75):
|