FitCheck / requirements.txt
cn0303's picture
Paste box goes live: fine-tuned spec parser fills the form from messy text
0e8e243 verified
# FitCheck — UI brick + deterministic engine + model brick (ZeroGPU).
gradio==6.16.0 # gr.Server (FastAPI) + @app.api queue + ZeroGPU glue
spaces # @spaces.GPU — ZeroGPU allocation on Hugging Face
torch>=2.8.0 # ZeroGPU requirement (>=2.8)
transformers>=5.8.0 # native NemotronH wired to hub kernels (>=5.8)
kernels>=0.12.0,<0.13 # transformers' own declared range — 0.15.x broke
# LayerRepository (revision became mandatory).
# Pulls kernels-community/mamba-ssm + causal-conv1d
# prebuilt for torch>=2.8 — replaces the naive Mamba
# path (pip mamba-ssm wheels stop at torch 2.7 and
# ZeroGPU build containers have no nvcc; never add
# mamba-ssm/causal-conv1d here, the build will fail)
accelerate # device placement / efficient loading
einops # required by the kernels-community mamba-ssm kernel
skops # safe loading of the trained speed predictor
xgboost # the speed predictor's runtime (engine/speed.py)
peft # spec-parser LoRA adapter loading