Fix pickle.dump missing file argument typo in README (#1316)

pytorch · Nov 20, 2024 · b7930ac · b7930ac
1 parent 8bc9046
commit b7930ac
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchao/quantization/README.md b/torchao/quantization/README.md
@@ -77,7 +77,7 @@ import torchao.quantization
 # After the first forward pass (when quantization was done)
 from torchao.quantization.autoquant import AUTOQUANT_CACHE
 with open("quantization-cache.pkl", "wb") as f:
-    pickle.dump(AUTOQUANT_CACHE)
+    pickle.dump(AUTOQUANT_CACHE, f)
 
 # On load
 from torchao.quantization.autoquant import AUTOQUANT_CACHE