Fix autogaze_model_id: bfshi/AutoGaze no longer exists
#9
by Cianmcnally - opened
- preprocessor_config.json +1 -1
- processing_nvila.py +1 -1
preprocessor_config.json
CHANGED
|
@@ -24,7 +24,7 @@
|
|
| 24 |
"height": 392,
|
| 25 |
"width": 392
|
| 26 |
},
|
| 27 |
-
"autogaze_model_id": "
|
| 28 |
"gazing_ratio_tile": 0.75,
|
| 29 |
"gazing_ratio_thumbnail": 0.75,
|
| 30 |
"task_loss_requirement_tile": 0.7,
|
|
|
|
| 24 |
"height": 392,
|
| 25 |
"width": 392
|
| 26 |
},
|
| 27 |
+
"autogaze_model_id": "nvidia/AutoGaze",
|
| 28 |
"gazing_ratio_tile": 0.75,
|
| 29 |
"gazing_ratio_thumbnail": 0.75,
|
| 30 |
"task_loss_requirement_tile": 0.7,
|
processing_nvila.py
CHANGED
|
@@ -142,7 +142,7 @@ class NVILAProcessor(ProcessorMixin):
|
|
| 142 |
self.tokenizer: Qwen2Tokenizer | Qwen2TokenizerFast
|
| 143 |
|
| 144 |
# AutoGaze configuration
|
| 145 |
-
self.autogaze_model_id = autogaze_model_id or "
|
| 146 |
self.gazing_ratio_tile = gazing_ratio_tile
|
| 147 |
self.gazing_ratio_thumbnail = gazing_ratio_thumbnail
|
| 148 |
self.task_loss_requirement_tile = task_loss_requirement_tile
|
|
|
|
| 142 |
self.tokenizer: Qwen2Tokenizer | Qwen2TokenizerFast
|
| 143 |
|
| 144 |
# AutoGaze configuration
|
| 145 |
+
self.autogaze_model_id = autogaze_model_id or "nvidia/AutoGaze"
|
| 146 |
self.gazing_ratio_tile = gazing_ratio_tile
|
| 147 |
self.gazing_ratio_thumbnail = gazing_ratio_thumbnail
|
| 148 |
self.task_loss_requirement_tile = task_loss_requirement_tile
|