persistworld / config.json
jaibrdhn's picture
Create config.json
864d0ba verified
Raw
History Blame Contribute Delete
602 Bytes
{
"model_type": "persistworld",
"architecture": "svd_action_conditioned_world_model",
"base_pipeline": "stabilityai/stable-video-diffusion-img2vid-xt",
"finetuned_from": "yjguo/Ctrl-World",
"checkpoint_file": "checkpoint-5760-merged.pt",
"checkpoint_contents": "merged UNet + action adapter weights",
"prediction_type": "sample",
"training_data": "cadene/droid_1.0.1",
"post_training": {
"method": "contrastive_denoising_rl",
"branching_factor": 16,
"trained_modules": ["lora_adapters", "action_encoder"],
"rewards": ["lpips", "ssim", "psnr"]
},
"license": "mit"
}