debug

trainer.py

@@ -73,10 +73,10 @@ class Trainer:
         remove_gpu_after_training: bool,
         input_token: str,
     ) -> str:
-        if SPACE_ID == ORIGINAL_SPACE_ID:
-            raise gr.Error(
-                'This Space does not work on this Shared UI. Duplicate the Space and attribute a GPU'
-            )
+        # if SPACE_ID == ORIGINAL_SPACE_ID:
+        #     raise gr.Error(
+        #         'This Space does not work on this Shared UI. Duplicate the Space and attribute a GPU'
+        #     )
         if not torch.cuda.is_available():
             raise gr.Error('CUDA is not available.')
         if training_video is None: