Skip to content

Conversation

@dguido
Copy link
Member

@dguido dguido commented Jan 23, 2026

Summary

  • Adds --huggingface REPO_ID argument to scan models directly from HuggingFace Hub
  • Adds --hf-revision and --hf-token for revision specification and private repos
  • Adds huggingface optional dependency: pip install fickling[huggingface]
  • Automatically filters for pickle-based files (.bin, .pt, .pth, .pkl, .pickle)
  • Skips known-safe formats (safetensors, json, txt, etc.)

Usage

# Scan a public model
fickling --huggingface bert-base-uncased --print-results

# Scan specific revision
fickling --huggingface bert-base-uncased --hf-revision main --print-results

# Scan private repo
fickling --huggingface my-org/private-model --hf-token $HF_TOKEN --print-results

Test plan

  • All existing tests pass
  • Linters pass
  • Manual testing with public HuggingFace models

🤖 Generated with Claude Code

- Add --huggingface REPO_ID argument to scan models directly from HuggingFace Hub
- Add --hf-revision and --hf-token arguments for specific revisions and private repos
- Add huggingface optional dependency (huggingface_hub >= 0.20.0)
- Automatically filter for pickle-based files and skip safe formats like safetensors

Usage: fickling --huggingface bert-base-uncased --print-results

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
@dguido dguido requested a review from ESultanik as a code owner January 23, 2026 02:50
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