xtinkt 1 yıl önce
ebeveyn
işleme
10de34b72a
1 değiştirilmiş dosya ile 1 ekleme ve 0 silme
  1. 1 0
      src/petals/server/block_functions.py

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

@@ -161,6 +161,7 @@ async def iterate_rpc_inference(
 
     async for request, step_metadata in input_iterator:
         if "last_validated_position" in step_metadata:
+            last_validated_position = step_metadata["last_validated_position"]
             assert prefix_length >= last_validated_position, f"prefix_length={prefix_length}, last_validated_position={last_validated_position}"
             prefix_length = last_validated_position