Spaces:
Sleeping
Sleeping
Commit
·
3faf794
1
Parent(s):
91d7875
SUPERIOR DEFAULT CPU
Browse files
app.py
CHANGED
@@ -32,7 +32,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
|
|
32 |
import argparse
|
33 |
|
34 |
parser = argparse.ArgumentParser('LocalGPT falcon', add_help=False)
|
35 |
-
parser.add_argument('--device_type', type=str, default="
|
36 |
args = parser.parse_args()
|
37 |
|
38 |
|
|
|
32 |
import argparse
|
33 |
|
34 |
parser = argparse.ArgumentParser('LocalGPT falcon', add_help=False)
|
35 |
+
parser.add_argument('--device_type', type=str, default="cpu", choices=["cpu", "mps", "cuda"], help='device type', )
|
36 |
args = parser.parse_args()
|
37 |
|
38 |
|