Sfoglia il codice sorgente

Update src/client/remote_sequential.py

justheuristic 3 anni fa
parent
commit
052116adb6
1 ha cambiato i file con 3 aggiunte e 0 eliminazioni
  1. 3 0
      src/client/remote_sequential.py

+ 3 - 0
src/client/remote_sequential.py

@@ -94,3 +94,6 @@ class RemoteTransformerBlock(RemoteSequential):
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
         assert len(self) == 1, "Remote Block is a sequence size 1"
+        
+    def extra_repr(self):
+        return f"{self.sequence_manager.block_uids[0]}"