o
    ?߱ik4                     @   s:   d dl Z d dlmZmZmZmZmZmZ e jdd Z	dS )    N)CheckpointConfigCheckpointDirHfCheckpointDirS3CheckpointFileHfCheckpointFileS3register_checkpointc                  C   sn  ddl m}  |   ttddddddtd	d
tddddd tdddd}ttddddddtdd
|d ttdddtdd
tddddd ttdddtd d
tdd!ddd ttd"d#d$dddtd%d
tddd&dd ttd'd(d)dd*d+d,d-td.d
tdd/d0dd ttd1d(d2dd*d+d,d-td3d
tddd4dd ttd5d6d7dddtd8d
tdd9d:dd ttd;d<d=d>d?dtd@d
tddAdBdd ttdCdDdEdddtdFd
tddGdHdd ttdIdJdKdLdddMtdNd
tddOdPdd ttdQdRdSdTdddMtdUd
tdVdWdXdd ttdYdZdSdTdddMtd[d
tdVd\d]dd d S )^Nr   )register_checkpointsz$d20b7120-df3e-4911-919d-db6e08bad31cz,nvidia/Cosmos-Predict2.5-2B/base/pre-trainedzStage-c_pt_4-reason_embeddings-v1p1-Index-26-Size-2B-Res-720-Fps-16-Note-T2V_high_sigma_loss_reweighted_1_1_rectified_flow_only_resume2Z720p   )
resolutionfpszs3://bucket/cosmos_diffusion_v2/official_runs_vid2vid/Stage-c_pt_4-reason_embeddings-v1p1-Index-26-Size-2B-Res-720-Fps-16-Note-T2V_high_sigma_loss_reweighted_1_1_rectified_flow_only_resume2/checkpoints/iter_000023000/model)uriznvidia/Cosmos-Predict2.5-2BZ(15a82a2ec231bc318692aa0456a36537c806e7d4zAbase/pre-trained/d20b7120-df3e-4911-919d-db6e08bad31c_ema_bf16.pt)
repositoryrevisionfilename)uuidname
experimentmetadatas3hfZ(f176dc95b4a70f53ce01c4b302851595e7322b00z5base/pre-trained/308eb96c-c4c0-4a06-9cc1-103a43beff28)r   r   Zsubdirectoryz$308eb96c-c4c0-4a06-9cc1-103a43beff28zgStage-c_pt_4-reason_embeddings-v1p1-Index-26-Size-2B-Res-720-Fps-16-Note-T2V_high_sigma_loss_reweightedzs3://bucket/cosmos_diffusion_v2/official_runs_text2world/Stage-c_pt_4-reason_embeddings-v1p1-Index-26-Size-2B-Res-720-Fps-16-Note-T2V_high_sigma_loss_reweighted/checkpoints/iter_000010000/modelz$7bbc8d06-2bc9-448d-94ee-b48b4ab7189cz'nvidia/Cosmos-Predict2.5-2B/interactiveZNcosmos_predict2p5_2B_action_conditioned_gr00t_gr1_customized_13frame_sf_warmupzs3://bucket/cosmos_predict2_action_conditioned/action_conditional/cosmos_predict2p5_2B_action_conditioned_gr00t_gr1_customized_13frame_full_16nodes/checkpoints/iter_000014000/modelznvidia/Cosmos-ExperimentalZ(2b5e9a99b58d5a61259ca99962c4c74127481006z67bbc8d06-2bc9-448d-94ee-b48b4ab7189c/model_ema_bf16.pt)r   r   r   r   r   z$bedc35da-1a54-4144-83db-6072c29b0fd9Z,cosmos_predict2p5_2B_action_gr00t_gr1_warmupzfs3://bucket/cosmos_predict2_action_conditioned/interactive_warmup/gr1/checkpoints/iter_000020000/modelZ(ded876a5b2e19aef64cd9d1100c03e5b05cf2f9cz$81edfebe-bd6a-4039-8c1d-737df1a790bfz-nvidia/Cosmos-Predict2.5-2B/base/post-trainedzAStage-c_pt_4-Index-2-Size-2B-Res-720-Fps-16-Note-rf_with_edm_ckptzs3://bucket/cosmos_diffusion_v2/official_runs_vid2vid/Stage-c_GRPO-reason_embeddings-Index-26-Size-2B-Res-720-Fps-16-posttrain_data-HQ_V7_RF_MERGE_LOCAL_ag_every2_guidance0_scorekeyoverall_reward_databeta0.01_mincon0/checkpoints/iter_000000288/modelzBbase/post-trained/81edfebe-bd6a-4039-8c1d-737df1a790bf_ema_bf16.ptz$524af350-2e43-496c-8590-3646ae1325daz*nvidia/Cosmos-Predict2.5-2B/auto/multiviewZbuttercup_predict2p5_2b_7views_res720p_fps30_t8_joint_alpamayo1capviewprefix_allcapsviewprefix_29frames_nofps_uniform_dropoutt0         )r
   r   viewsframeszs3://bucket/cosmos_predict2_multiview/cosmos2_mv/buttercup_predict2p5_2b_7views_res720p_fps30_t8_joint_alpamayo1capviewprefix_allcapsviewprefix_29frames_nofps_uniform_dropoutt0-0/checkpoints/iter_000012000/modelZ(865baf084d4c9e850eac59a021277d5a9b9e8b63z?auto/multiview/524af350-2e43-496c-8590-3646ae1325da_ema_bf16.ptz$6b9d7548-33bb-4517-b5e8-60caf47edba7Zbuttercup_predict2p5_2b_7views_res720p_fps30_t8_from48kfps30mv_condprobs0442_joint_alpamayo1capnoviewprefix_allcapsviewprefix_29frames_nofpszs3://bucket/cosmos_predict2_multiview/cosmos2_mv/buttercup_predict2p5_2b_7views_res720p_fps30_t8_from48kfps30mv_condprobs0442_joint_alpamayo1capnoviewprefix_allcapsviewprefix_29frames_nofps-0/checkpoints/iter_000005000/modelz?auto/multiview/6b9d7548-33bb-4517-b5e8-60caf47edba7_ema_bf16.ptz$f740321e-2cd6-4370-bbfe-545f4eca2065z2nvidia/Cosmos-Predict2.5-2B/robot/multiview-agibotZKmulticamera_video2video_rectified_flow_2b_res_720_fps16_s3_agibot_frameinitzs3://bucket/cosmos_diffusion_v2/official_runs_vid2vid/multicamera_video2video_rectified_flow_2b_res_720_fps16_s3_agibot_frameinit/checkpoints/iter_000016500Z(fbe72c18d152053029a19db3b211cf78671ad422zGrobot/multiview-agibot/f740321e-2cd6-4370-bbfe-545f4eca2065_ema_bf16.ptz$38c6c645-7d41-4560-8eeb-6f4ddc0e6574z-nvidia/Cosmos-Predict2.5-2B/robot/action-condZ_cosmos_predict2p5_2B_reason_embeddings_action_conditioned_rectified_flow_bridge_13frame_256x320Z360p   zs3://bucket/cosmos_predict2_action_conditioned/action_conditional/cosmos_predict2p5_2B_reason_embeddings_action_conditioned_rectified_flow_bridge_13frame_256x320/checkpoints/iter_000016000/modelmainzBrobot/action-cond/38c6c645-7d41-4560-8eeb-6f4ddc0e6574_ema_bf16.ptz$24a3b7b8-6a3d-432d-b7d1-5d30b9229465z'nvidia/Cosmos-Predict2.5-2B/transfer2.5zStage-c_pt_4-reason_embeddings-v1p1-Index-26-Size-2B-Res-720-Fps-16-Note-T2V_high_sigma_loss_reweighted_1_1_rectified_flow_onlyzs3://bucket/cosmos_diffusion_v2/official_runs_text2world/Stage-c_pt_4-reason_embeddings-v1p1-Index-26-Size-2B-Res-720-Fps-16-Note-T2V_high_sigma_loss_reweighted_1_1_rectified_flow_only/checkpoints/iter_000037000/modelZ(9a02ed8daa8c6c7718ac09da06488bfd1d363cb6z624a3b7b8-6a3d-432d-b7d1-5d30b9229465/model_ema_bf16.ptz$575edf0f-d973-4c74-b52c-69929a08d0a5z*nvidia/Cosmos-Predict2.5-2B/base/distilledZ<dmd2_trigflow_distill_cosmos_predict2_2B_bidirectional_TnI2V2B)sizer
   r   zs3://bucket/cosmos_predict2_distill/predict2_distill/dmd2_trigflow_distill_cosmos_predict2_2B_bidirectional/checkpoints/iter_000007500/modelZ(e26f8a125a2235c5a00245a65207402dd0cdcb89z?base/distilled/575edf0f-d973-4c74-b52c-69929a08d0a5_ema_bf16.ptz$54937b8c-29de-4f04-862c-e67b04ec41e8z-nvidia/Cosmos-Predict2.5-14B/base/pre-trainedz{Stage-c_pt_4-reason_embeddings-v1p1-Index-43-Size-14B-Res-720-Fps-16_resume_from_reason1p1_rectified_flow_shift5_high_sigma14Bzs3://bucket/cosmos_diffusion_v2/official_runs_text2world/Stage-c_pt_4-reason_embeddings-v1p1-Index-43-Size-14B-Res-720-Fps-16_resume_from_reason1p1_rectified_flow_shift5_high_sigma/checkpoints/iter_000012500/modelznvidia/Cosmos-Predict2.5-14BZ(03eb354f35eae0d6e0c1be3c9f94d8551e125570zAbase/pre-trained/54937b8c-29de-4f04-862c-e67b04ec41e8_ema_bf16.ptz$e21d2a49-4747-44c8-ba44-9f6f9243715fz.nvidia/Cosmos-Predict2.5-14B/base/post-traineda  s3://bucket/cosmos_diffusion_v2/official_runs_vid2vid/Stage-c_GRPO-reason_embeddings-Index-26-Size-14B-Res-720-Fps-16-posttrain_data-HQ_V7_RF_MERGE_GENERAL_steps20_every2_lr3e-6_guidance0_scorekeyoverall_reward_databeta0.01_mincon0/checkpoints/iter_000000128/modelZ(2bc4ca5ba5a20b9858a7ddb856bc82d70b030fbezBbase/post-trained/e21d2a49-4747-44c8-ba44-9f6f9243715f_ema_bf16.pt)Zcosmos_oss.checkpointsr   r   r   r   r   r   r   )Z_register_checkpointsZcheckpoint_hf r    ]/data/cameron/vidgen/cosmos-predict2.5/packages/cosmos-oss/cosmos_oss/checkpoints_predict2.pyr      s  r   )
	functools3cosmos_predict2._src.imaginaire.utils.checkpoint_dbr   r   r   r   r   r   cacher   r    r    r    r!   <module>   s    
