vahe1994 1 éve
szülő
commit
43751d50bd
1 módosított fájl, 1 hozzáadás és 0 törlés
  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)