Spaces:
Build error
Build error
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 |
|