Spaces:
Sleeping
Sleeping
Commit
·
5bceece
1
Parent(s):
671d7f9
Add stage='default' parameter to oneshot() call
Browse files- Add stage parameter as mentioned in instructions
- All parameters now correctly map to HfArgumentParser dataclasses
- Recipe.from_modifiers() and Dataset.from_dict() already in place
quantize_to_awq_colab.ipynb
CHANGED
|
@@ -375,11 +375,13 @@
|
|
| 375 |
" # - num_calibration_samples: DatasetArguments.num_calibration_samples\n",
|
| 376 |
" # - use_auth_token: ModelArguments.use_auth_token (reads from HF_TOKEN env var)\n",
|
| 377 |
" # - trust_remote_code_model: ModelArguments.trust_remote_code_model\n",
|
|
|
|
| 378 |
" print(f\" → Calling oneshot() with proper argument structure...\")\n",
|
| 379 |
" oneshot(\n",
|
| 380 |
" model=repo_id,\n",
|
| 381 |
" output_dir=temp_output_dir,\n",
|
| 382 |
" recipe=recipe,\n",
|
|
|
|
| 383 |
" dataset=calibration_dataset,\n",
|
| 384 |
" num_calibration_samples=min(calibration_dataset_size, len(calibration_dataset)),\n",
|
| 385 |
" use_auth_token=True, # Reads from os.environ[\"HF_TOKEN\"]\n",
|
|
|
|
| 375 |
" # - num_calibration_samples: DatasetArguments.num_calibration_samples\n",
|
| 376 |
" # - use_auth_token: ModelArguments.use_auth_token (reads from HF_TOKEN env var)\n",
|
| 377 |
" # - trust_remote_code_model: ModelArguments.trust_remote_code_model\n",
|
| 378 |
+
" # - stage: RecipeArguments.stage (default: \"default\")\n",
|
| 379 |
" print(f\" → Calling oneshot() with proper argument structure...\")\n",
|
| 380 |
" oneshot(\n",
|
| 381 |
" model=repo_id,\n",
|
| 382 |
" output_dir=temp_output_dir,\n",
|
| 383 |
" recipe=recipe,\n",
|
| 384 |
+
" stage=\"default\", # Recipe stage\n",
|
| 385 |
" dataset=calibration_dataset,\n",
|
| 386 |
" num_calibration_samples=min(calibration_dataset_size, len(calibration_dataset)),\n",
|
| 387 |
" use_auth_token=True, # Reads from os.environ[\"HF_TOKEN\"]\n",
|