Spaces:
Runtime error
Runtime error
gurgenblbulyan
commited on
Commit
•
48e8be6
1
Parent(s):
7ccc14a
Update inference.py
Browse files- 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=
|
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)
|