Dmlで動かなくなってたのを修正 (#30)

Hiroshiba · Dec 21, 2024 · 5788dd3 · 5788dd3
1 parent 2433ab0
commit 5788dd3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vv_core_inference/make_yukarin_sosoa_forwarder.py b/vv_core_inference/make_yukarin_sosoa_forwarder.py
@@ -124,7 +124,7 @@ def forward(
 
         h = self.pre(h)
 
-        mask = torch.ones_like(f0).squeeze()
+        mask = torch.ones_like(f0)[:, :, 0]
         h, _ = self.encoder(h, None, mask)
 
         output1 = self.post(h)