Spaces:

huggan
/

sefa

Runtime error

Johannes Kolbe commited on Apr 15, 2022

Commit

424816e

•

1 Parent(s): f5fc044

remove cuda() call

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -28,7 +28,7 @@ def to_tensor(array):
       A `torch.Tensor` with dtype `torch.FloatTensor` on cuda device.
     """
     assert isinstance(array, np.ndarray)
-    return torch.from_numpy(array).type(torch.FloatTensor).cuda()
 def postprocess(images, min_val=-1.0, max_val=1.0):

       A `torch.Tensor` with dtype `torch.FloatTensor` on cuda device.
     """
     assert isinstance(array, np.ndarray)
+    return torch.from_numpy(array).type(torch.FloatTensor)
 def postprocess(images, min_val=-1.0, max_val=1.0):