diff --git a/spaces.py b/spaces.py index 01ccb9ba..6c10a3c6 100644 --- a/spaces.py +++ b/spaces.py @@ -128,7 +128,7 @@ def capture_gpu_object(): return GPUObject() -def GPU(gpu_objects=None, manual_load=False): +def GPU(gpu_objects=None, manual_load=False, **kwargs): gpu_objects = gpu_objects or [] if not isinstance(gpu_objects, (list, tuple)):