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

Fix SDXL LoRa offloading and SD 1.5 parsing

Browse Source
This commit is contained in:
Disty0
2023-08-10 16:02:45 +03:00
parent 0b5dd46d79
commit 0a7105d524
3 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/prompt_parser_diffusers.py
+5 -5
View File
@@ -3,7 +3,6 @@ import typing
import torch
import diffusers
from compel import Compel, ReturnedEmbeddingsType
import modules.devices as devices
import modules.shared as shared
import modules.prompt_parser as prompt_parser
@@ -55,9 +54,10 @@ def compel_encode_prompts(
negative_pooleds.append(negative_pooled)
prompt_embeds = torch.cat(prompt_embeds, dim=0)
positive_pooleds = torch.cat(positive_pooleds, dim=0)
negative_embeds = torch.cat(negative_embeds, dim=0)
negative_pooleds = torch.cat(negative_pooleds, dim=0)
if shared.sd_model_type == "sdxl":
positive_pooleds = torch.cat(positive_pooleds, dim=0)
negative_pooleds = torch.cat(negative_pooleds, dim=0)
return prompt_embeds, positive_pooleds, negative_embeds, negative_pooleds
@@ -94,7 +94,7 @@ def compel_encode_prompt(
text_encoder=pipeline.text_encoder,
returned_embeddings_type=embedding_type,
requires_pooled=False,
device=devices.device
device=shared.device
)
if shared.sd_model_type == "sdxl":
@@ -103,7 +103,7 @@ def compel_encode_prompt(
text_encoder=pipeline.text_encoder_2,
returned_embeddings_type=embedding_type,
requires_pooled=True,
device=devices.device
device=shared.device
)
if not is_refiner:
positive_te1 = compel_te1(prompt)