Fix default args

mikekgfb · web-flow · commit 3956f53950ca · 2024-10-14T14:39:17.000-07:00
Fix default args
diff --git a/torchchat/utils/quantize.py b/torchchat/utils/quantize.py
@@ -585,7 +585,7 @@ class WeightOnlyInt8QuantHandler(QuantHandler):
     def __init__(
         self,
         model: Optional[nn.Module] = None,
-        device,
+        device = None,
         precision=None,
         tokenizer=None,
         *,
