Card: code links and usage for the budgetpix package layout

README.md

@@ -16,7 +16,7 @@ tags:
 
 # BudgetPix-pMF
 
-🌐 [Project page](https://budgetany.github.io/budgetpix/)  |  💻 [Code](https://github.com/BudgetAny/BudgetPix/tree/main/pmf)  |  📄 [Paper](https://arxiv.org/abs/2610.12307)  |  🧩 [BudgetAny](https://huggingface.co/BudgetAny)
+🌐 [Project page](https://budgetany.github.io/budgetpix/)  |  💻 [Code](https://github.com/BudgetAny/BudgetPix/tree/main/budgetpix/pmf)  |  📄 [Paper](https://arxiv.org/abs/2610.12307)  |  🧩 [BudgetAny](https://huggingface.co/BudgetAny)
 
 **One checkpoint, any token budget.** One-step class-conditional ImageNet generation in pixel space with [pixel MeanFlow (pMF)](https://github.com/Lyy-iiis/pMF), fine-tuned with BudgetPix compute-adaptive tokenization. At inference you choose how many tokens the transformer runs on: an entropy-based detail map merges the flattest regions into coarse tokens first and gives the fine tokens to the most detailed ones; the layout is chosen up front, by default from the model's own dense sample of the same class and noise. The same weights serve every budget.
 
@@ -37,9 +37,9 @@ tags:
 ## Usage
 
 ```bash
-git clone https://github.com/BudgetAny/BudgetPix && cd BudgetPix/pmf    # setup: see README.md
+git clone https://github.com/BudgetAny/BudgetPix && cd BudgetPix    # setup: see README.md
 hf download BudgetAny/BudgetPix-pMF budgetpix_pmf_l16_256.pt --local-dir weights
-python sample.py --ckpt weights/budgetpix_pmf_l16_256.pt --budget 128 --classes 207,88,417,933
+python -m budgetpix.pmf.sample --ckpt weights/budgetpix_pmf_l16_256.pt --budget 128 --classes 207,88,417,933
 ```
 
 `--budget` is the number of tokens per image (256 is the dense grid).