vahe1994 1 年之前
父节点
当前提交
43751d50bd
共有 1 个文件被更改,包括 1 次插入0 次删除
  1. 1 0
      src/petals/server/from_pretrained.py

+ 1 - 0
src/petals/server/from_pretrained.py

@@ -81,6 +81,7 @@ def load_pretrained_block(
         cache_dir=cache_dir,
         max_disk_space=max_disk_space,
     )
+    print(block)
     print(model_name, block_prefix, revision,token,cache_dir,max_disk_space)
 
     # print("now printing", state_dict)