Skip to content

Commit b68c0bd

Browse files
committed
Fix checkpoint interval
1 parent 893e73c commit b68c0bd

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

conversion/measure.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -387,7 +387,7 @@ def measure_quant(job, save_fn, model):
387387
overall_rolling_accuracy = 0
388388

389389
last_snapshot_time = time.time()
390-
snapshot_interval_s = 90
390+
snapshot_interval_s = 180
391391

392392
temp_filename = os.path.join(job["out_dir"], "hidden_states_temp.safetensors")
393393
states_filename = os.path.join(job["out_dir"], "hidden_states.safetensors")

conversion/quantize.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -253,7 +253,7 @@ def quant_parallel_decoder(job, module, hidden_states, target_states, quantizers
253253
def quant(job, save_fn, model):
254254

255255
last_snapshot_time = time.time()
256-
snapshot_interval_s = 90
256+
snapshot_interval_s = 180
257257

258258
temp_filename = os.path.join(job["out_dir"], "hidden_states_temp.safetensors")
259259
states_filename = os.path.join(job["out_dir"], "hidden_states.safetensors")
@@ -526,4 +526,4 @@ def quant(job, save_fn, model):
526526
del job["invalid"]
527527
save_fn()
528528

529-
time_since_snapshot = time.time()
529+
last_snapshot_time = time.time()

0 commit comments

Comments
 (0)