MaxReimann
commited on
Commit
β’
3a4b39d
1
Parent(s):
15d8bee
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]
|