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
@ssss141414 ssss141414 added the model-scale-by-skill Model support PR created or maintained by the adding-model-support skill label Aug 12, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

model-scale-by-skill Model support PR created or maintained by the adding-model-support skill

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants