remove legacy config

This commit is contained in:
layerdiffusion
2024-08-03 17:04:56 -07:00
parent fb3052350b
commit be3f0a0039
3 changed files with 3 additions and 3 deletions

View File

@@ -350,7 +350,7 @@ class ControlLora(ControlNet):
def pre_run(self, model, percent_to_timestep_function):
super().pre_run(model, percent_to_timestep_function)
controlnet_config = model.diffusion_model.legacy_config.copy()
controlnet_config = model.diffusion_model.config.copy()
controlnet_config.pop("out_channels")
controlnet_config["hint_channels"] = self.control_weights["input_hint_block.0.weight"].shape[1]
controlnet_config["dtype"] = dtype = model.storage_dtype