Spaces:
Runtime error
Runtime error
set_lora_device
Browse files- scripts/process_utils.py +1 -1
scripts/process_utils.py
CHANGED
@@ -102,7 +102,7 @@ def initialize_sotai_model():
|
|
102 |
|
103 |
for lora_name, alpha in lora_names:
|
104 |
lora_path = get_file_path(lora_name, subfolder=os.environ["lora_dir"])
|
105 |
-
load_lora(sotai_gen_pipe, lora_path, alpha)
|
106 |
|
107 |
# スケジューラーの設定
|
108 |
sotai_gen_pipe.scheduler = UniPCMultistepScheduler.from_config(sotai_gen_pipe.scheduler.config)
|
|
|
102 |
|
103 |
for lora_name, alpha in lora_names:
|
104 |
lora_path = get_file_path(lora_name, subfolder=os.environ["lora_dir"])
|
105 |
+
load_lora(sotai_gen_pipe, lora_path, adapter_name=lora_name, alpha=alpha)
|
106 |
|
107 |
# スケジューラーの設定
|
108 |
sotai_gen_pipe.scheduler = UniPCMultistepScheduler.from_config(sotai_gen_pipe.scheduler.config)
|