Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 17:24:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix tensor conversion

Browse Source
This commit is contained in:
Vladimir Mandic
2023-12-11 10:22:53 -05:00
parent b7cbd34106
commit 0d805e3ed7
2 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing.py
+7 -1
View File
@@ -757,7 +757,13 @@ def validate_sample(tensor):
return tensor
if tensor.dtype == torch.bfloat16: # numpy does not support bf16
tensor = tensor.to(torch.float16)
sample = tensor.detach().cpu().numpy() if isinstance(tensor, torch.Tensor) and hasattr(tensor, 'detach') else tensor.cpu().numpy()
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):
sample = tensor
else:
shared.log.warning(f'Unknown sample type: {type(tensor)}')
sample = 255.0 * np.moveaxis(sample, 0, 2) if shared.backend == shared.Backend.ORIGINAL else 255.0 * sample
with warnings.catch_warnings(record=True) as w:
cast = sample.astype(np.uint8)