kamp/IOPaint
1
0
Fork 0
Code Issues Pull Requests Releases 1 Activity

fix lcm_lora load

Browse Source
This commit is contained in:
Qing 2024-01-08 21:49:27 +08:00
parent 5da47ee035
commit a1a06004d8
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
iopaint/model_manager.py
View File

@ -166,8 +166,11 @@ class ModelManager:
def enable_disable_lcm_lora(self, config: InpaintRequest): def enable_disable_lcm_lora(self, config: InpaintRequest):
if self.available_models[self.name].support_lcm_lora: if self.available_models[self.name].support_lcm_lora:
# TODO: change this if load other lora is supported
lcm_lora_loaded = bool(self.model.model.get_list_adapters())
if config.sd_lcm_lora: if config.sd_lcm_lora:
if not self.model.model.get_list_adapters(): if not lcm_lora_loaded:
self.model.model.load_lora_weights(self.model.lcm_lora_id) self.model.model.load_lora_weights(self.model.lcm_lora_id)
else: else:
if lcm_lora_loaded:
self.model.model.disable_lora() self.model.model.disable_lora()