Prepwork for ilora

This commit is contained in:
Jaret Burkett
2024-01-12 06:41:15 -07:00
parent 290393f7ae
commit e190fbaeb8
3 changed files with 13 additions and 10 deletions

View File

@@ -288,6 +288,9 @@ class IPAdapter(torch.nn.Module):
output_dim=sd.unet.config['cross_attention_dim'],
ff_mult=4
)
elif adapter_config.type == 'ilora':
# we apply the clip encodings to the LoRA
image_proj_model = None
else:
raise ValueError(f"unknown adapter type: {adapter_config.type}")