lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-02-28 02:33:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

make results more consistent to A1111

Browse Source
This commit is contained in:
layerdiffusion
2024-08-08 01:53:03 -07:00
parent e396307e9d
commit a05a06b337
3 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
backend/text_processing/t5_engine.py
View File

@@ -58,7 +58,7 @@ class T5TextProcessingEngine:
return tokenized
def encode_with_transformers(self, tokens):
device = memory_management.get_torch_device()
device = memory_management.text_encoder_device()
tokens = tokens.to(device)
self.text_encoder.shared.to(device=device, dtype=torch.float32)