Remove missed arg

Jack-Khuu · Jack-Khuu · commit a44d22b6f025 · 2024-09-22T01:06:45.000-07:00
diff --git a/torchchat/cli/builder.py b/torchchat/cli/builder.py
@@ -532,7 +532,7 @@ def _initialize_model(
         # ), "quantize not valid for exported PTE model. Specify quantization during export."
 
         with measure_time("Time to load model: {time:.02f} seconds"):
-            model = _load_model(builder_args, only_config=True)
+            model = _load_model(builder_args)
             device_sync(device=builder_args.device)
 
         try:
