o
    ?߱i	                     @   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   sl   t tddtddtdddd t td	d
tddtddddd t tddtddtdddd d S )Nz$7219c6c7-f878-4137-bbdb-76842ea85e70zQwen/Qwen2.5-VL-7B-InstructzSs3://bucket/cosmos_reasoning1/pretrained/Qwen_tokenizer/Qwen/Qwen2.5-VL-7B-Instruct)uriznvidia/Cosmos-Reason1-7BZ(3210bec0495fdc7a8d3dbb8d58da5711eab4b423)
repositoryrevision)uuidnames3hfz$685afcaa-4de2-42fe-b7b9-69f7a2dee4d8z
Wan2.1/vaezQs3://bucket/cosmos_diffusion_v2/pretrain_weights/tokenizer/wan2pt1/Wan2.1_VAE.pthznvidia/Cosmos-Predict2.5-2BZ(6787e176dce74a101d922174a95dba29fa5f0c55ztokenizer.pth)r	   r
   filenamez$cb3e3ffa-7b08-4c34-822d-61c7aa31a14fznvidia/Cosmos-Reason1.1-7Bzs3://bucket/cosmos_reasoning1/sft_exp700/sft_exp721-1_qwen7b_tl_721_5vs5_s3_balanced_n32_resume_16k/checkpoints/iter_000016000/model)r   r   r   r   r   r    r   r   T/data/cameron/vidgen/cosmos-predict2.5/packages/cosmos-oss/cosmos_oss/checkpoints.pyregister_checkpoints   sP   r   )
	functools3cosmos_predict2._src.imaginaire.utils.checkpoint_dbr   r   r   r   r   r   cacher   r   r   r   r   <module>   s    
