darkstar94 commited on
Commit
d021954
1 Parent(s): b92567e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -23,7 +23,7 @@ from transformers import AutoModel, AutoTokenizer
23
  # Argparser
24
  parser = argparse.ArgumentParser(description='demo')
25
  #parser.add_argument('--device', type=str, default='mps', help='cuda or mps')
26
- parser.add_argument('--dtype', type=str, default='fp16', help='bf16 or fp16')
27
  args = parser.parse_args()
28
  #device = args.device
29
  #assert device in ['cuda', 'mps']
 
23
  # Argparser
24
  parser = argparse.ArgumentParser(description='demo')
25
  #parser.add_argument('--device', type=str, default='mps', help='cuda or mps')
26
+ parser.add_argument('--dtype', type=str, default='bf16', help='bf16 or fp16')
27
  args = parser.parse_args()
28
  #device = args.device
29
  #assert device in ['cuda', 'mps']