oliveryanzuolu commited on
Commit
5c928d2
·
1 Parent(s): 175fdf1

Add RAVEN demo and checkpoint links

Browse files
Files changed (1) hide show
  1. README.md +8 -4
README.md CHANGED
@@ -27,14 +27,18 @@ CM-GRPO is the reinforcement-learning stage on top of RAVEN. It treats a consist
27
 
28
  This repository hosts weights only. Training, inference, and evaluation code live in the [code repository](https://github.com/mvp-ai-lab/RAVEN). Other artifacts are listed in the [RAVEN collection](https://huggingface.co/collections/mvp-lab/raven).
29
 
 
 
 
 
30
  ## Files
31
 
32
  | File | Description |
33
  | --- | --- |
34
- | `raven_model.pt` | RAVEN backbone (post-distillation, pre-RL). Full `CausalWanModel` state dict. |
35
- | `cmgrpo_raven_merge.pt` | CM-GRPO backbone with the LoRA already folded into RAVEN. Same key set as `raven_model.pt`. |
36
- | `cmgrpo_raven_lora.safetensors` | CM-GRPO adapter only. Use together with `raven_model.pt` as the base. |
37
- | `cmgrpo_raven_full.pt` | PEFT-wrapped bundle of base + adapter, kept as an archival interchange format. The current code cannot load it directly. Convert or merge it offline first. |
38
 
39
  ## Model details
40
 
 
27
 
28
  This repository hosts weights only. Training, inference, and evaluation code live in the [code repository](https://github.com/mvp-ai-lab/RAVEN). Other artifacts are listed in the [RAVEN collection](https://huggingface.co/collections/mvp-lab/raven).
29
 
30
+ ## Demo
31
+
32
+ <video controls loop muted playsinline preload="metadata" width="100%" src="https://github.com/user-attachments/assets/628a070c-fee5-4b77-97af-5875e79fca1d"></video>
33
+
34
  ## Files
35
 
36
  | File | Description |
37
  | --- | --- |
38
+ | [Wan2.1-T2V-1.3B-RAVEN](https://huggingface.co/mvp-lab/RAVEN/blob/main/raven_model.pt) | RAVEN backbone (post-distillation, pre-RL). Full `CausalWanModel` state dict. |
39
+ | [Wan2.1-T2V-1.3B-CMGRPO-Merge](https://huggingface.co/mvp-lab/RAVEN/blob/main/cmgrpo_raven_merge.pt) | CM-GRPO backbone with the LoRA already folded into RAVEN. Same key set as `raven_model.pt`. |
40
+ | [Wan2.1-T2V-1.3B-CMGRPO-LoRA](https://huggingface.co/mvp-lab/RAVEN/blob/main/cmgrpo_raven_lora.safetensors) | CM-GRPO adapter only. Use together with `raven_model.pt` as the base. |
41
+ | [Wan2.1-T2V-1.3B-CMGRPO-Full](https://huggingface.co/mvp-lab/RAVEN/blob/main/cmgrpo_raven_full.pt) | PEFT-wrapped bundle of base + adapter, kept as an archival interchange format. The current code cannot load it directly. Convert or merge it offline first. |
42
 
43
  ## Model details
44