yeq6x commited on
Commit
c3251dc
·
1 Parent(s): d81c6ca

set_lora_device

Browse files
Files changed (1) hide show
  1. 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)