KangLiao commited on
Commit
03c345f
·
1 Parent(s): 2788c5c
configs/models/qwen2_5_1_5b_radio_sd3_dynamic_puffin.py CHANGED
@@ -84,9 +84,7 @@ model = dict(type=Qwen2p5RadioStableDiffusion3HFDynamic,
84
  pretrained_pth=None,
85
  use_activation_checkpointing=False,
86
  visual_encoder=dict(
87
- type=RADIOModel.from_config,
88
- pretrained_model_name_or_path=radiov3_model_name_or_path,
89
- torch_dtype=torch.bfloat16,
90
- #local_files_only=True,
91
- ),
92
  )
 
84
  pretrained_pth=None,
85
  use_activation_checkpointing=False,
86
  visual_encoder=dict(
87
+ type=RADIOModel.from_pretrained,
88
+ pretrained_model_name_or_path="nvidia/C-RADIOv3-H",
89
+ torch_dtype=torch.bfloat16,),
 
 
90
  )