init
Browse files
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.
|
| 88 |
-
pretrained_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 |
)
|