EricaCorral commited on
Commit
fe7b5d7
1 Parent(s): c2b09ef

increase performance by disabling dropout/batchnorm layers

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -5,6 +5,7 @@ import gradio as gr
5
  import torch
6
 
7
  model = M2M100ForConditionalGeneration.from_pretrained("facebook/m2m100_1.2B")
 
8
  tokenizer = M2M100Tokenizer.from_pretrained("facebook/m2m100_1.2B")
9
  lac = LAC(mode="seg")
10
 
 
5
  import torch
6
 
7
  model = M2M100ForConditionalGeneration.from_pretrained("facebook/m2m100_1.2B")
8
+ model.eval()
9
  tokenizer = M2M100Tokenizer.from_pretrained("facebook/m2m100_1.2B")
10
  lac = LAC(mode="seg")
11