nodes/mcDandy-more_math
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

(AI) fix

Browse Source
This commit is contained in:
mcDandy
2026-01-25 14:49:55 +01:00
parent c7237b36fd
commit 2794623194
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
more_math/Parser/UnifiedMathVisitor.py
+5 -1
View File
@@ -159,7 +159,11 @@ class UnifiedMathVisitor(MathExprVisitor):
true_t = self._promote_to_tensor(true_val)
false_t = self._promote_to_tensor(false_val)
cond_t = torch.isclose(condition, torch.tensor(0.0, device=self.device)) == False
if true_t.dtype != false_t.dtype:
true_t = true_t.float()
false_t = false_t.float()
cond_t = torch.isclose(condition.float(), torch.tensor(0.0, device=self.device)) == False
return torch.where(cond_t, true_t, false_t).contiguous()
if self._is_list(condition):