Procházet zdrojové kódy

check for absent args

justheuristic před 3 roky
rodič
revize
8ff45d136d
1 změnil soubory, kde provedl 1 přidání a 1 odebrání
  1. 1 1
      src/client/remote_block.py

+ 1 - 1
src/client/remote_block.py

@@ -35,7 +35,7 @@ class RemoteTransformerBlock(RemoteExpert):
 
     def forward(self, inputs: torch.Tensor, **kwargs):
         for k, v in kwargs.items():
-            assert v is None, f"Extra keyword arguments are not yet supported (got {k} = {v})"
+            assert v is None or v is False, f"Extra keyword arguments are not yet supported (got {k} = {v})"
         return super().forward(inputs)
 
     def inference_session(self) -> RemoteTransformerBlockInferenceSession: