tight-inversion commited on
Commit
addfbd1
·
1 Parent(s): d672aed

Use fp8 model

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -453,6 +453,7 @@ if __name__ == "__main__":
453
  parser.add_argument("--pretrained_model", type=str, help='for development')
454
  args = parser.parse_args()
455
 
 
456
  if args.aggressive_offload:
457
  args.offload = True
458
 
 
453
  parser.add_argument("--pretrained_model", type=str, help='for development')
454
  args = parser.parse_args()
455
 
456
+ args.fp8 = True
457
  if args.aggressive_offload:
458
  args.offload = True
459