Fix Minor Issue with distribute argument

spcl · Jul 22, 2024 · 13555c7 · 13555c7
1 parent 6aa2da2
commit 13555c7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/fake_quant/main.py b/fake_quant/main.py
@@ -149,7 +149,7 @@ def main():
 
 
 
-    if args.distribute_model:
+    if args.distribute:
         utils.distribute_model(model)
     else:
         model.to(utils.DEV)
@@ -169,4 +169,4 @@ def main():
 
 
 if __name__ == '__main__':
-    main()
+    main()