Alikestocode commited on
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

Files changed (1) hide show
  1. quantize_to_awq_colab.ipynb +2 -0
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",