yeq6x commited on
Commit
a742794
1 Parent(s): 4c75bd8

set_lora_device

Browse files
Files changed (1) hide show
  1. scripts/process_utils.py +3 -3
scripts/process_utils.py CHANGED
@@ -100,9 +100,9 @@ def initialize_sotai_model():
100
  # (os.environ["lora_name2"], 0.3),
101
  ]
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.split(".")[0], alpha=alpha)
106
 
107
  # スケジューラーの設定
108
  sotai_gen_pipe.scheduler = UniPCMultistepScheduler.from_config(sotai_gen_pipe.scheduler.config)
 
100
  # (os.environ["lora_name2"], 0.3),
101
  ]
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.split(".")[0], alpha=alpha)
106
 
107
  # スケジューラーの設定
108
  sotai_gen_pipe.scheduler = UniPCMultistepScheduler.from_config(sotai_gen_pipe.scheduler.config)