|
44 | 44 | from PIL import Image |
45 | 45 | from safetensors.torch import load_file |
46 | 46 |
|
47 | | -from transformers import EomtDinov3Config, EomtDinov3ForUniversalSegmentation, EomtImageProcessorFast |
| 47 | +from transformers import EomtDinov3Config, EomtDinov3ForUniversalSegmentation, EomtImageProcessor |
48 | 48 |
|
49 | 49 |
|
50 | 50 | CAT_URL = "http://images.cocodataset.org/val2017/000000039769.jpg" |
@@ -340,7 +340,7 @@ def convert_model( |
340 | 340 |
|
341 | 341 | model.load_state_dict(merged_state_dict, strict=True, assign=True) |
342 | 342 |
|
343 | | - processor = EomtImageProcessorFast( |
| 343 | + processor = EomtImageProcessor( |
344 | 344 | size={"shortest_edge": image_size, "longest_edge": image_size}, |
345 | 345 | do_split_image=False, |
346 | 346 | do_pad=True, |
@@ -373,7 +373,7 @@ def convert_model( |
373 | 373 | processor.push_to_hub(repo_id=f"nielsr/{repo_name}") |
374 | 374 |
|
375 | 375 |
|
376 | | -def _prepare_image(processor: EomtImageProcessorFast) -> torch.Tensor: |
| 376 | +def _prepare_image(processor: EomtImageProcessor) -> torch.Tensor: |
377 | 377 | image = Image.open(requests.get(CAT_URL, stream=True).raw).convert("RGB") |
378 | 378 | inputs = processor(images=image, do_normalize=False, return_tensors="pt") |
379 | 379 | return inputs.pixel_values |
@@ -573,7 +573,7 @@ def _assert_allclose(reference: Iterable[torch.Tensor], actual: Iterable[torch.T |
573 | 573 | def verify_conversion( |
574 | 574 | *, |
575 | 575 | hf_model: EomtDinov3ForUniversalSegmentation, |
576 | | - processor: EomtImageProcessorFast, |
| 576 | + processor: EomtImageProcessor, |
577 | 577 | delta_state_dict: dict[str, torch.Tensor], |
578 | 578 | backbone_repo_id: str, |
579 | 579 | image_size: int, |
|
0 commit comments