ZeroGPU

BidirectionalTranslation/options/base_options.py

@@ -64,7 +64,8 @@ class BaseOptions():
         # Special tasks
         self.initialized = True
         return parser
-
+        
+    @spaces.GPU
     def gather_options(self):
         """Initialize our parser with basic options (only once)."""
         if not self.initialized:
@@ -114,6 +115,7 @@ class BaseOptions():
             opt_file.write(message)
             opt_file.write('\n')
 
+    @spaces.GPU
     def parse(self, model_global_path):
         """Parse options, create checkpoints directory suffix, and set up gpu device."""
         self.model_global_path = model_global_path