Răsfoiți Sursa

check for absent args

justheuristic 3 ani în urmă
părinte
comite
8ff45d136d
1 a modificat fișierele cu 1 adăugiri și 1 ștergeri
  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: