瀏覽代碼

Update src/client/remote_sequential.py

justheuristic 3 年之前
父節點
當前提交
052116adb6
共有 1 個文件被更改,包括 3 次插入0 次删除
  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]}"