File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed
Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -614,12 +614,12 @@ def download(
614614 )
615615 url += "&checkpoint=model_output_path"
616616 elif isinstance (ft_job .training_type , LoRATrainingType ):
617- if checkpoint_type == DownloadCheckpointType .DEFAULT :
618- checkpoint_type = DownloadCheckpointType .MERGED
617+ if checkpoint_type == DownloadCheckpointType .DEFAULT . value :
618+ checkpoint_type = DownloadCheckpointType .MERGED . value
619619
620- if checkpoint_type == DownloadCheckpointType .MERGED :
620+ if checkpoint_type == DownloadCheckpointType .MERGED . value :
621621 url += f"&checkpoint={ DownloadCheckpointType .MERGED .value } "
622- elif checkpoint_type == DownloadCheckpointType .ADAPTER :
622+ elif checkpoint_type == DownloadCheckpointType .ADAPTER . value :
623623 url += f"&checkpoint={ DownloadCheckpointType .ADAPTER .value } "
624624 else :
625625 raise ValueError (
You can’t perform that action at this time.
0 commit comments