Update README.md
Browse files
README.md
CHANGED
@@ -42,12 +42,12 @@ Here is how to use this model in PyTorch:
|
|
42 |
>>> model_id = "rmihaylov/gpt2-small-theseus-bg"
|
43 |
>>> tokenizer = AutoTokenizer.from_pretrained(model_id)
|
44 |
>>> model = AutoModel.from_pretrained(model_id, trust_remote_code=True)
|
45 |
-
|
46 |
>>> input_ids = tokenizer.encode(
|
47 |
>>> "Здравей,",
|
48 |
>>> add_special_tokens=False,
|
49 |
>>> return_tensors='pt')
|
50 |
-
|
51 |
>>> output_ids = model.generate(
|
52 |
>>> input_ids,
|
53 |
>>> do_sample=True,
|
@@ -55,7 +55,7 @@ Here is how to use this model in PyTorch:
|
|
55 |
>>> top_p=0.92,
|
56 |
>>> pad_token_id=2,
|
57 |
>>> top_k=0)
|
58 |
-
|
59 |
>>> output = tokenizer.decode(output_ids[0])
|
60 |
>>>
|
61 |
>>> output = output.replace('<|endoftext|>', '\n\n\n')
|
|
|
42 |
>>> model_id = "rmihaylov/gpt2-small-theseus-bg"
|
43 |
>>> tokenizer = AutoTokenizer.from_pretrained(model_id)
|
44 |
>>> model = AutoModel.from_pretrained(model_id, trust_remote_code=True)
|
45 |
+
>>>
|
46 |
>>> input_ids = tokenizer.encode(
|
47 |
>>> "Здравей,",
|
48 |
>>> add_special_tokens=False,
|
49 |
>>> return_tensors='pt')
|
50 |
+
>>>
|
51 |
>>> output_ids = model.generate(
|
52 |
>>> input_ids,
|
53 |
>>> do_sample=True,
|
|
|
55 |
>>> top_p=0.92,
|
56 |
>>> pad_token_id=2,
|
57 |
>>> top_k=0)
|
58 |
+
>>>
|
59 |
>>> output = tokenizer.decode(output_ids[0])
|
60 |
>>>
|
61 |
>>> output = output.replace('<|endoftext|>', '\n\n\n')
|