OpenPipe
diff --git a/‎src/art/megatron/service.py‎
Lines changed: 1 addition & 1 deletion b/‎src/art/megatron/service.py‎
Lines changed: 1 addition & 1 deletion
@@ -32,7 +32,7 @@
     DEFAULT_VLLM_WAKE_LOCK_PATH,
     MegatronTrainingJob,
 )
-from .shared import merge_lora_adapter
+from .train import merge_lora_adapter
 
 safetensors = importlib.import_module("safetensors")
 safe_open = safetensors.safe_open
Original file line number	Diff line number	Diff line change
`@@ -32,7 +32,7 @@`
`32`	`32`	`DEFAULT_VLLM_WAKE_LOCK_PATH,`
`33`	`33`	`MegatronTrainingJob,`
`34`	`34`	`)`
`35`		`-from .shared import merge_lora_adapter`
	`35`	`+from .train import merge_lora_adapter`
`36`	`36`
`37`	`37`	`safetensors = importlib.import_module("safetensors")`
`38`	`38`	`safe_open = safetensors.safe_open`