BK-Lee commited on
Commit
5d55bce
·
1 Parent(s): e1361b1
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ accel = Accelerator()
19
 
20
  # loading meteor model
21
  mmamba = load_mmamba('BK-Lee/Meteor-Mamba')
22
- meteor, tok_meteor = load_meteor('BK-Lee/Meteor-MLM', bits=16)
23
 
24
  # freeze model
25
  freeze_model(mmamba)
 
19
 
20
  # loading meteor model
21
  mmamba = load_mmamba('BK-Lee/Meteor-Mamba')
22
+ meteor, tok_meteor = load_meteor('BK-Lee/Meteor-MLM', bits=4)
23
 
24
  # freeze model
25
  freeze_model(mmamba)