Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-18 16:54:33 +02:00
Code Issues Packages Projects Releases Wiki Activity

add attention masks

Browse Source
This commit is contained in:
Enes Sadık Özbek
2025-06-23 01:27:17 +00:00
parent 5e9c6bfd4e
commit 954138bfc1
4 changed files with 124 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing_class.py
+4
View File
@@ -152,6 +152,8 @@ class StableDiffusionProcessing:
self.positive_pooleds = []
self.negative_embeds = []
self.negative_pooleds = []
self.prompt_attention_masks = []
self.negative_prompt_attention_masks = []
self.disable_extra_networks = False
self.iteration = 0
self.network_data = {}
@@ -321,6 +323,8 @@ class StableDiffusionProcessing:
self.positive_pooleds = []
self.negative_embeds = []
self.negative_pooleds = []
self.prompt_attention_masks = []
self.negative_prompt_attention_mask = []
def __str__(self):
return f'{self.__class__.__name__}: {self.__dict__}'