ready2drop commited on
Commit
516bce8
·
verified ·
1 Parent(s): de11d67
Files changed (1) hide show
  1. app.py +0 -3
app.py CHANGED
@@ -26,9 +26,6 @@ def parse_args(args):
26
  parser.add_argument('--scale', type=bool, default=True)
27
  parser.add_argument('--smote', type=bool, default=True)
28
  parser.add_argument('--model_name_or_path', type=str, default="./FINAL", choices=[])
29
- parser.add_argument('--top_p', type=float, default=None)
30
- parser.add_argument('--temperature', type=float, default=1.0)
31
- parser.add_argument('--device', type=str, default="cuda", choices=["cuda", "cpu"])
32
 
33
  return parser.parse_args(args)
34
 
 
26
  parser.add_argument('--scale', type=bool, default=True)
27
  parser.add_argument('--smote', type=bool, default=True)
28
  parser.add_argument('--model_name_or_path', type=str, default="./FINAL", choices=[])
 
 
 
29
 
30
  return parser.parse_args(args)
31