Fix autogaze_model_id: bfshi/AutoGaze no longer exists

#9
Files changed (2) hide show
  1. preprocessor_config.json +1 -1
  2. processing_nvila.py +1 -1
preprocessor_config.json CHANGED
@@ -24,7 +24,7 @@
24
  "height": 392,
25
  "width": 392
26
  },
27
- "autogaze_model_id": "bfshi/AutoGaze",
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 "bfshi/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
 
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