vahe1994 1 year ago
parent
commit
f6081bcdc9
1 changed files with 2 additions and 0 deletions
  1. 2 0
      src/petals/server/server.py

+ 2 - 0
src/petals/server/server.py

@@ -487,6 +487,8 @@ class ModuleContainer(threading.Thread):
         blocks = {}
         try:
             for module_uid, block_index in zip(module_uids, block_indices):
+                print(f'{converted_model_name_or_path=},{block_index=}')
+
                 block = load_pretrained_block(
                     converted_model_name_or_path,
                     block_index,