update

2025-08-04 11:12:35 +00:00 · 2024-02-22 20:17:10 +08:00
parent b7aa425344
commit ba66cf8d69
2 changed files with 2 additions and 1 deletions
--- a/modules/hypernetworks/hypernetwork.py
+++ b/modules/hypernetworks/hypernetwork.py
@@ -95,6 +95,7 @@ class HypernetworkModule(torch.nn.Module):
                        zeros_(b)
                    else:
                        raise KeyError(f"Key {weight_init} is not defined as initialization!")
+        devices.torch_npu_set_device()
        self.to(devices.device)

    def fix_old_state_dict(self, state_dict):