Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-08-27 07:31:01 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix tensor to np

Browse Source
This commit is contained in:
Vladimir Mandic
2023-12-10 18:06:07 -05:00
parent 4823595dc9
commit d71287e068
3 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
extensions-builtin/sd-webui-controlnet
+1 -1
Submodule extensions-builtin/sd-webui-controlnet updated: 96dbc601a6...0d0841671e
extensions-builtin/stable-diffusion-webui-images-browser
+1 -1
Submodule extensions-builtin/stable-diffusion-webui-images-browser updated: 08fc2647f1...323ce7efef
modules/processing.py
+1 -1
View File
@@ -757,7 +757,7 @@ def validate_sample(tensor):
sample = 255.0 * np.moveaxis(tensor.cpu().numpy(), 0, 2)
else:
sample = 255.0 * tensor
if isinstance(tensor, torch.Tensor):
if isinstance(tensor, torch.Tensor) and hasattr(tensor, 'detach'):
sample = sample.detach().cpu().numpy()
with warnings.catch_warnings(record=True) as w:
cast = sample.astype(np.uint8)