Adapt to ZeroGPU?

utils/geometry.py

@@ -524,8 +524,6 @@ def get_operators(verts, faces, k_eig=128, cache_path=None, normals=None, overwr
                 evecs = torch.from_numpy(evecs).to(device=device, dtype=dtype)
                 gradX = sparse_np_to_torch(gradX).to(device=device, dtype=dtype)
                 gradY = sparse_np_to_torch(gradY).to(device=device, dtype=dtype)
-                hks = torch.from_numpy(hks).to(device=device, dtype=dtype)
-                wks = torch.from_numpy(wks).to(device=device, dtype=dtype)
 
                 found = True
 
@@ -606,7 +604,6 @@ def load_operators(filepath):
         evecs=npzfile['evecs'],
         gradX=read_sp_mat('gradX'),
         gradY=read_sp_mat('gradY'),
-        hks=npzfile['hks'],
         wks=npzfile['wks'], 
     )