MaxReimann
commited on
Update 1_🎨_Apply_preset.py
Browse files
pages/1_🎨_Apply_preset.py
CHANGED
@@ -98,7 +98,7 @@ with coll1:
|
|
98 |
|
99 |
res_data = None
|
100 |
if canvas_result.image_data is not None:
|
101 |
-
abc = np_to_torch(canvas_result.image_data.astype(np.
|
102 |
|
103 |
img_org_width = org_cuda.shape[-1]
|
104 |
img_org_height = org_cuda.shape[-2]
|
|
|
98 |
|
99 |
res_data = None
|
100 |
if canvas_result.image_data is not None:
|
101 |
+
abc = np_to_torch(canvas_result.image_data.astype(np.float32)).sum(dim=1, keepdim=True).cuda()
|
102 |
|
103 |
img_org_width = org_cuda.shape[-1]
|
104 |
img_org_height = org_cuda.shape[-2]
|