Skip to content

Commit

Permalink
supports other format loras for unet
Browse files Browse the repository at this point in the history
  • Loading branch information
entrpn committed Dec 10, 2024
1 parent 5058538 commit f75de2f
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions src/maxdiffusion/maxdiffusion_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -54,6 +54,7 @@ def _noop_interceptor(next_fn, args, kwargs, context):
weight_name=lora_config["weight_name"][0],
params=params,
adapter_name=lora_config["adapter_name"][0],
unet_config=pipeline.unet.config
)
interceptor = pipeline.make_lora_interceptor(params, rank, network_alphas)

Expand Down

0 comments on commit f75de2f

Please sign in to comment.