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