nodes/nullquant-ComfyUI-BrushNet
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

lora scaling removed in prompt_encode()

Browse Source
This commit is contained in:
nullquant
2024-04-18 21:01:45 +03:00
parent 8bdd0f1996
commit dd4aa6c6b0
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
brushnet/pipeline_brushnet.py
+6 -3
View File
@@ -430,9 +430,12 @@ class StableDiffusionBrushNetPipeline(
negative_prompt_embeds = negative_prompt_embeds.repeat(1, num_images_per_prompt, 1)
negative_prompt_embeds = negative_prompt_embeds.view(batch_size * num_images_per_prompt, seq_len, -1)
if isinstance(self, LoraLoaderMixin) and USE_PEFT_BACKEND:
# Retrieve the original scale by scaling back the LoRA layers
unscale_lora_layers(self.text_encoder, lora_scale)
# self.text_encoder is not used after encode_prompt()
# self.text_encoder = None (we use prompt_embeds and negative_prompt_embeds in brushnet_nodes)
# So, nullquant commented out this section.
#if isinstance(self, LoraLoaderMixin) and USE_PEFT_BACKEND:
# # Retrieve the original scale by scaling back the LoRA layers
# unscale_lora_layers(self.text_encoder, lora_scale)
return prompt_embeds, negative_prompt_embeds