Cianmcnally commited on
Commit
5f02643
·
verified ·
1 Parent(s): 7a5670e

Fix autogaze_model_id: bfshi/AutoGaze no longer exists

Browse files

Loading this model's processor currently fails because it tries to pull AutoGaze from `bfshi/AutoGaze`, which isn't on the Hub anymore (looks like it used to be on @bfshi 's personal account before moving to `nvidia/AutoGaze`):

```
RepositoryNotFoundError: 404 Client Error.
Repository Not Found for url: https://huggingface.co/api/models/bfshi/AutoGaze
```

Changed `autogaze_model_id` in `preprocessor_config.json` and the fallback in `processing_nvila.py` to `nvidia/AutoGaze`. Until now I've had to patch these two files locally every time I use the model.

Same problem in the AutoGaze quick start docs, fix open here: https://github.com/NVlabs/AutoGaze/pull/6

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