itsmeadarsh commited on
Commit
3faf794
1 Parent(s): 91d7875

SUPERIOR DEFAULT CPU

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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="cuda", choices=["cpu", "mps", "cuda"], help='device type', )
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