Browse Source

check for absent args

justheuristic 3 năm trước cách đây
mục cha
commit
8ff45d136d
1 tập tin đã thay đổi với 1 bổ sung1 xóa
  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: