Spaces:

AlexZou
/

Deploy_Restoration

Build error

AlexZou commited on Nov 17, 2022

Commit

8961161

1 Parent(s): ebe6a67

Update Lowlight.py

Files changed (1) hide show

Lowlight.py CHANGED Viewed

@@ -23,7 +23,7 @@ input_image = torch.from_numpy(input_image).float()
 input_image = input_image.permute(2, 0, 1).unsqueeze(0)
 input_image = input_image.numpy()
-providers = ['CUDAExecutionProvider', 'CPUExecutionProvider']
 model_name = 'IAT'
 print('-' * 50)

 input_image = input_image.permute(2, 0, 1).unsqueeze(0)
 input_image = input_image.numpy()
+providers = ['CPUExecutionProvider']
 model_name = 'IAT'
 print('-' * 50)