vahe1994 hai 1 ano
pai
achega
25ac2214ca
Modificáronse 1 ficheiros con 2 adicións e 0 borrados
  1. 2 0
      src/petals/server/server.py

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

@@ -514,6 +514,7 @@ class ModuleContainer(threading.Thread):
                     cache_dir=cache_dir,
                     max_disk_space=max_disk_space,
                 )
+                print("finished with block converssion")
                 blocks[module_uid] = TransformerBackend(
                     module_uid,
                     block,
@@ -535,6 +536,7 @@ class ModuleContainer(threading.Thread):
                     min_batch_size=min_batch_size,
                     max_batch_size=max_batch_size,
                 )
+                print("finished with TransformerBackend")
 
             merge_inference_pools_inplace(blocks)