fix:lora not use different checkpoint model when optional_lora_stack in xyplot #113

This commit is contained in:
yolain
2024-04-02 13:22:54 +08:00
parent 1e1b82399c
commit 84307b95aa
2 changed files with 5 additions and 0 deletions
+2
View File
@@ -327,6 +327,8 @@ class imageRatio:
height_ratio = height // ratio
result = (width_ratio, height_ratio, width_ratio, height_ratio)
else:
width_ratio = 0
height_ratio = 0
result = (0, 0, 0.0, 0.0)
text = f"Image Ratio is {str(width_ratio)}:{str(height_ratio)}"
+3
View File
@@ -308,6 +308,7 @@ class easyXYPlot():
ckpt_name, clip_skip, vae_name = xy_values.split(",")
ckpt_name = ckpt_name.replace('*', ',')
vae_name = vae_name.replace('*', ',')
print(ckpt_name)
model, clip, vae, clip_vision = self.easyCache.load_checkpoint(ckpt_name)
if vae_name != 'None':
vae = self.easyCache.load_vae(vae_name)
@@ -316,6 +317,8 @@ class easyXYPlot():
optional_lora_stack = plot_image_vars['lora_stack']
if optional_lora_stack is not None and optional_lora_stack != []:
for lora in optional_lora_stack:
lora['model'] = model
lora['clip'] = clip
model, clip = self.easyCache.load_lora(lora)
# 处理clip