Update README.md
Browse files
README.md
CHANGED
@@ -6,7 +6,7 @@ library_name: transformers
|
|
6 |
pipeline_tag: text2text-generation
|
7 |
inference:
|
8 |
parameters:
|
9 |
-
max_new_tokens:
|
10 |
widget:
|
11 |
- text: >-
|
12 |
Ben buraya bazı <MASK> istiyorum.
|
@@ -38,10 +38,6 @@ tokenizer = AutoTokenizer.from_pretrained("vngrs-ai/VBART-Medium-Base",
|
|
38 |
# Uncomment the device_map kwarg and delete the closing bracket to use model for inference on GPU
|
39 |
model = AutoModelForSeq2SeqLM.from_pretrained("vngrs-ai/VBART-Medium-Base")#, device_map="auto")
|
40 |
|
41 |
-
context="..."
|
42 |
-
question="..."
|
43 |
-
highlighted_context="..."
|
44 |
-
|
45 |
# Input text
|
46 |
input_text = "Ben buraya bazı <MASK> istiyorum."
|
47 |
|
|
|
6 |
pipeline_tag: text2text-generation
|
7 |
inference:
|
8 |
parameters:
|
9 |
+
max_new_tokens: 250
|
10 |
widget:
|
11 |
- text: >-
|
12 |
Ben buraya bazı <MASK> istiyorum.
|
|
|
38 |
# Uncomment the device_map kwarg and delete the closing bracket to use model for inference on GPU
|
39 |
model = AutoModelForSeq2SeqLM.from_pretrained("vngrs-ai/VBART-Medium-Base")#, device_map="auto")
|
40 |
|
|
|
|
|
|
|
|
|
41 |
# Input text
|
42 |
input_text = "Ben buraya bazı <MASK> istiyorum."
|
43 |
|