lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-03-06 05:29:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

hard code clip token device

Browse Source 
#950
This commit is contained in:
layerdiffusion
2024-08-08 01:36:01 -07:00
parent a189f3e53e
commit e396307e9d
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
backend/text_processing/classic_engine.py
View File

@@ -120,7 +120,7 @@ class ClassicTextProcessingEngine:
return tokenized
def encode_with_transformers(self, tokens):
target_device = memory_management.get_torch_device()
target_device = self.text_encoder.transformer.text_model.embeddings.token_embedding.weight.device
self.text_encoder.transformer.text_model.embeddings.position_ids = self.text_encoder.transformer.text_model.embeddings.position_ids.to(device=target_device)
tokens = tokens.to(target_device)