Unverified Commit 589931ca authored by Tolga Cangöz's avatar Tolga Cangöz Committed by GitHub
Browse files

Errata - Update class method convention to use `cls` (#8574)



* Class methods are supposed to use `cls` conventionally

* `make style && make quality`

* An Empty commit

---------
Co-authored-by: default avatarSayak Paul <spsayakpaul@gmail.com>
parent 675be88f
......@@ -2165,7 +2165,7 @@ class SDXLLongPromptWeightingPipeline(
@classmethod
def save_lora_weights(
self,
cls,
save_directory: Union[str, os.PathLike],
unet_lora_layers: Dict[str, Union[torch.nn.Module, torch.Tensor]] = None,
text_encoder_lora_layers: Dict[str, Union[torch.nn.Module, torch.Tensor]] = None,
......@@ -2188,7 +2188,7 @@ class SDXLLongPromptWeightingPipeline(
state_dict.update(pack_weights(text_encoder_lora_layers, "text_encoder"))
state_dict.update(pack_weights(text_encoder_2_lora_layers, "text_encoder_2"))
self.write_lora_layers(
cls.write_lora_layers(
state_dict=state_dict,
save_directory=save_directory,
is_main_process=is_main_process,
......
......@@ -1339,7 +1339,7 @@ class DemoFusionSDXLPipeline(
@classmethod
def save_lora_weights(
self,
cls,
save_directory: Union[str, os.PathLike],
unet_lora_layers: Dict[str, Union[torch.nn.Module, torch.Tensor]] = None,
text_encoder_lora_layers: Dict[str, Union[torch.nn.Module, torch.Tensor]] = None,
......@@ -1368,7 +1368,7 @@ class DemoFusionSDXLPipeline(
state_dict.update(pack_weights(text_encoder_lora_layers, "text_encoder"))
state_dict.update(pack_weights(text_encoder_2_lora_layers, "text_encoder_2"))
self.write_lora_layers(
cls.write_lora_layers(
state_dict=state_dict,
save_directory=save_directory,
is_main_process=is_main_process,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment