gurgenblbulyan commited on
Commit
48e8be6
1 Parent(s): 7ccc14a

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -4,7 +4,7 @@ from transformers import AutoTokenizer, VisionEncoderDecoderModel
4
  import utils
5
 
6
  class Inference:
7
- def __init__(self, decoder_model_name, max_length=32):
8
  self.device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
9
 
10
  self.tokenizer = AutoTokenizer.from_pretrained(decoder_model_name)
 
4
  import utils
5
 
6
  class Inference:
7
+ def __init__(self, decoder_model_name, max_length=50):
8
  self.device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
9
 
10
  self.tokenizer = AutoTokenizer.from_pretrained(decoder_model_name)