hoelee/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-08-04 11:12:35 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #14981 from wangshuai09/gpu_info_for_ascend

Browse Source 
Add training support and change lspci for Ascend NPU
This commit is contained in:
AUTOMATIC1111
2024-03-04 20:06:54 +03:00
committed by GitHub
parent 09b5ce68a9 3fb1c2e58d
commit eee46a5094
3 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
modules/sd_hijack_clip.py
View File

@@ -230,7 +230,7 @@ class FrozenCLIPEmbedderWithCustomWordsBase(torch.nn.Module):
for fixes in self.hijack.fixes:
for _position, embedding in fixes:
used_embeddings[embedding.name] = embedding
devices.torch_npu_set_device()
z = self.process_tokens(tokens, multipliers)
zs.append(z)