vahe1994 hace 1 año
padre
commit
43751d50bd
Se han modificado 1 ficheros con 1 adiciones y 0 borrados
  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)