OPENGIT/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-04-30 00:39:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
wangshuai09 2024-02-22 20:17:10 +08:00
parent b7aa425344
commit ba66cf8d69
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
modules/hypernetworks/hypernetwork.py
View File

@ -95,6 +95,7 @@ class HypernetworkModule(torch.nn.Module):
zeros_(b) zeros_(b)
else: else:
raise KeyError(f"Key {weight_init} is not defined as initialization!") raise KeyError(f"Key {weight_init} is not defined as initialization!")
devices.torch_npu_set_device()
self.to(devices.device) self.to(devices.device)
def fix_old_state_dict(self, state_dict): def fix_old_state_dict(self, state_dict):

2
modules/sd_hijack_clip.py
View File

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