Update main.py

f6e7ae25 · Leo Gao · GitHub · af0b1e26 · f6e7ae25
Unverified Commit f6e7ae25 authored Jun 25, 2021 by Leo Gao Committed by GitHub Jun 25, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

main.py main.py +1 -1

No files found.
--- a/main.py
+++ b/main.py
@@ -16,7 +16,7 @@ def parse_args():
    parser.add_argument('--provide_description', action="store_true")
    parser.add_argument('--num_fewshot', type=int, default=0)
    parser.add_argument('--batch_size', type=int, default=None)
-    parser.add_argument('--device', type=int, default=None)
+    parser.add_argument('--device', type=str, default=None)
    parser.add_argument('--seed', type=int, default=1234)
    parser.add_argument('--output_path', default=None)
    parser.add_argument('--limit', type=int, default=None)