Update train_dpo load_state call to handle result (#65)

andriigrynenko · web-flow · commit d19c3d5023c9 · 2025-10-31T15:08:05.000-07:00
diff --git a/tinker_cookbook/preference/train_dpo.py b/tinker_cookbook/preference/train_dpo.py
@@ -97,7 +97,7 @@ def create_dpo_clients(
     )
     if load_state_path:
         # Load state into the training client
-        training_client.load_state(load_state_path)
+        training_client.load_state(load_state_path).result()
         logger.info(f"Loaded weights from {load_state_path}")
     # Create a sampling client for the reference model from the training client
     reference_client = training_client.save_weights_and_get_sampling_client("reference")

Original file line number	Diff line number	Diff line change
`@@ -97,7 +97,7 @@ def create_dpo_clients(`
`97`	`97`	`)`
`98`	`98`	`if load_state_path:`
`99`	`99`	`# Load state into the training client`
`100`		`- training_client.load_state(load_state_path)`
	`100`	`+ training_client.load_state(load_state_path).result()`
`101`	`101`	`logger.info(f"Loaded weights from {load_state_path}")`
`102`	`102`	`# Create a sampling client for the reference model from the training client`
`103`	`103`	`reference_client = training_client.save_weights_and_get_sampling_client("reference")`