Skip to content

Add a multitask architecture module and its model sources #1256

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
priorphil merged 10 commits into from
Sep 10, 2026
Merged

Add a multitask architecture module and its model sources #1256

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
10 commits
Select commit Hold shift + click to select a range
f79b05a
Add TabPFN-3.5 and TabPFN-3.5-Fast as model versions v3.5 and v3.5-fast
bejaeger Sep 9, 2026
796e82c
create_default_for_version: take a ModelVersion, as the classifier does
bejaeger Sep 9, 2026
203e0be
Point the v3.5 model sources at the 20260909 exports and list the mul…
Sep 9, 2026
44253a0
v3.5: fill missing train cells with the cached path's fill value; com…
Sep 9, 2026
57cdd42
Drop the tabpfn_v3_5_prerelease alias: the released checkpoints name …
Sep 9, 2026
b7b324e
Drop the changelog fragment; the release notes come with the default …
Sep 9, 2026
42109c5
Say that TabPFN-3.5 is not available yet when its repo is not visible…
Sep 9, 2026
634ccd7
Restore the direct base architecture alias check
priorphil Sep 10, 2026
4807ecd
Reuse the validated class count in the v3.5 decoder
priorphil Sep 10, 2026
136f190
Keep the decoder equivalence test independent of full-graph compilation
priorphil Sep 10, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 2 additions & 1 deletion src/tabpfn/architectures/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@

from typing import TYPE_CHECKING

from . import tabpfn_v2, tabpfn_v2_5, tabpfn_v2_6, tabpfn_v3
from . import tabpfn_v2, tabpfn_v2_5, tabpfn_v2_6, tabpfn_v3, tabpfn_v3_5

if TYPE_CHECKING:
from tabpfn.architectures.interface import ArchitectureModule
Expand All @@ -23,6 +23,7 @@
"tabpfn_v2_5": tabpfn_v2_5,
"tabpfn_v2_6": tabpfn_v2_6,
"tabpfn_v3": tabpfn_v3,
"tabpfn_v3_5": tabpfn_v3_5,
}
"""Map from architecture names to the corresponding module."""

Expand Down
Loading