rmihaylov commited on
Commit
533baf0
1 Parent(s): ad0bdad

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
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')