Skip to content

feat(modelkit): add swin2sr support and canonical cpu recipe#1204

Open
fangyangci wants to merge 2 commits into
mainfrom
fangyangci/add-caidas-swin2sr-support
Open

feat(modelkit): add swin2sr support and canonical cpu recipe#1204
fangyangci wants to merge 2 commits into
mainfrom
fangyangci/add-caidas-swin2sr-support

Conversation

@fangyangci

Copy link
Copy Markdown
Contributor

Summary

  • Add Swin2SR task/class registration and ONNX registration shim
  • Wire Swin2SR mapping into HF aggregate registry
  • Add canonical recipe at examples/recipes/caidas_swin2SR-classical-sr-x2-64/cpu/cpu/image-to-image_fp16_config.json
  • Add unit tests for swin2sr support and loader task coverage

Validation

  • uv run pytest tests/unit/models/test_swin2sr_support.py tests/unit/loader/test_get_supported_tasks.py -q
  • uv run python -c "import json; from pathlib import Path; from winml.modelkit.config import WinMLBuildConfig; p=Path('examples/recipes/caidas_swin2SR-classical-sr-x2-64/cpu/cpu/image-to-image_fp16_config.json'); d=json.loads(p.read_text(encoding='utf-8')); WinMLBuildConfig.from_dict(d); print('recipe-parse-ok')"
  • uv run winml inspect --model-type swin2sr --list-tasks

@fangyangci
fangyangci requested a review from a team as a code owner July 24, 2026 05:22

from transformers import Swin2SRConfig

import winml.modelkit.models # noqa: F401 # trigger registrations
@fangyangci
fangyangci enabled auto-merge (squash) July 24, 2026 05:49
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants