nodes/blepping-comfyui_overly_complicated_sampling
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix generator expression error w/ list comprehension (#7)

Browse Source 
* Fix incompatibility with older Python versions in expression flip function handler
This commit is contained in:
zer0int
2024-08-27 14:43:48 -06:00
committed by GitHub
parent f1a8decc55
commit e58189e5f0
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
py/expression_handlers.py
+3 -3
View File
@@ -102,11 +102,11 @@ class FlipHandler(NormHandler):
return torch.flip(tensor, (dim,))
result = tensor.detach().clone()
pivot = tensor.shape[dim] // 2
out_slice = (
out_slice = tuple(
np.s_[:] if d != dim else np.s_[pivot:] for d in range(tensor.ndim)
)
in_slice = (np.s_[:] if d != dim else np.s_[:pivot] for d in range(tensor.ndim))
result[*out_slice] = torch.flip(tensor[*in_slice], dims=(dim,))
in_slice = tuple(np.s_[:] if d != dim else np.s_[:pivot] for d in range(tensor.ndim))
result[out_slice] = torch.flip(tensor[in_slice], dims=(dim,))
return result