annabeth97c commited on
Commit
3a8e6af
·
verified ·
1 Parent(s): 50a2c91

Update default model

Browse files
src/sonicverse/multi_token/training.py CHANGED
@@ -115,7 +115,7 @@ class ModelArguments:
115
  modality_builder: str = field(default="vision_clip")
116
  use_multi_task: int = field(default=MultiTaskType.PROJECTED_MULTI_TASK)
117
  tasks_config: str = field(default="src/sonicverse/configs/tasks.json")
118
- model_lora_path: Optional[str] = field(default="annabeth97c/sonicverse")
119
 
120
 
121
  class LMMTrainer(Trainer):
 
115
  modality_builder: str = field(default="vision_clip")
116
  use_multi_task: int = field(default=MultiTaskType.PROJECTED_MULTI_TASK)
117
  tasks_config: str = field(default="src/sonicverse/configs/tasks.json")
118
+ model_lora_path: Optional[str] = field(default="amaai-lab/SonicVerse")
119
 
120
 
121
  class LMMTrainer(Trainer):