google-research / google-research/tabfm

PyPI release (1.0.0) is incompatible with the published Hugging Face checkpoint(for kaggle); loader fix exists on main but has not been released

Open
#77 1 comment 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
2.6k
Forks
270
Avg merge
1d 7h
Merged PRs (30d)
1

Description

The current PyPI release (tabfm==1.0.0) cannot load the official Hugging Face PyTorch checkpoint (google/tabfm-1.0.0-pytorch).

After investigating the issue, it appears that this is a release mismatch rather than an architecture or checkpoint problem. The current GitHub source already contains a rewritten loader (TabFM_HF.from_pretrained() via PyTorchModelHubMixin), but PyPI still distributes the older loader implementation.

The current PyPI package (tabfm==1.0.0) cannot load the official
google/tabfm-1.0.0-pytorch checkpoint.

The published Hugging Face repository contains

classification/
model.safetensors

while the released PyPI package expects

classification/pytorch_model.bin

and calls torch.load().

This causes

FileNotFoundError:
Weights not found at ...

I noticed the current repository already contains a rewritten loader
based on PyTorchModelHubMixin (version reports 1.0.1 in the source tree),
which correctly delegates loading through Hugging Face.

Therefore, this appears to be a packaging/release mismatch rather than
an implementation bug.

The GitHub source contains the fix, but the published package used by
Kaggle and pip users still contains the older loader.

Would it be possible to publish a new PyPI release matching the current
repository so users can load the official checkpoint without patching
the loader manually?

Contributor guide

Open the contributing guide

Research direction

Compare the PyPI tabfm==1.0.0 package with the current source, especially TabFM_HF.from_pretrained() and the source version reported as 1.0.1. Check the project’s packaging and release workflow, then verify that a new PyPI release loads google/tabfm-1.0.0-pytorch from Hugging Face in the Kaggle or pip use case. Done means the published package matches the current loader and no longer expects the missing pytorch_model.bin file.

Written by the indexing model from the issue text.

Assessment

Tech stack
huggingface, python, pytorch
Domain
machine-learning, release
Issue type
Bug
Difficulty
3/5
Estimated time
1-2 days
Activity status
Quiet
Clarity
Mostly clear
Newbie friendliness
55/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.