王浩东 commited on
Commit ·
8da092d
1
Parent(s): aeeaf0b
roberta
Browse files
mvggt/models/mvggt_training.py
CHANGED
|
@@ -250,7 +250,7 @@ class MVGGT(nn.Module):
|
|
| 250 |
ckpt=None,
|
| 251 |
pretrained_model_name_or_path=None,
|
| 252 |
use_referring_segmentation=False,
|
| 253 |
-
text_model_name='
|
| 254 |
freeze_visual_modules=False,
|
| 255 |
use_masked_attn=True,
|
| 256 |
num_multimodal_layers=12,
|
|
|
|
| 250 |
ckpt=None,
|
| 251 |
pretrained_model_name_or_path=None,
|
| 252 |
use_referring_segmentation=False,
|
| 253 |
+
text_model_name='roberta-base',
|
| 254 |
freeze_visual_modules=False,
|
| 255 |
use_masked_attn=True,
|
| 256 |
num_multimodal_layers=12,
|