Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-11 15:28:43 +02:00
Code Issues Packages Projects Releases Wiki Activity

cleanup

Browse Source
This commit is contained in:
Vladimir Mandic
2023-12-11 10:56:41 -05:00
parent 0fe27642a5
commit 839e12e062
1 changed files with 0 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing.py
-1
View File
@@ -757,7 +757,6 @@ def validate_sample(tensor):
return tensor
if tensor.dtype == torch.bfloat16: # numpy does not support bf16
tensor = tensor.to(torch.float16)
print('HERE', isinstance(tensor, np.ndarray), isinstance(tensor, torch.Tensor))
if isinstance(tensor, torch.Tensor) and hasattr(tensor, 'detach'):
sample = tensor.detach().cpu().numpy()
elif isinstance(tensor, np.ndarray):