Arabic-NLP / test.py
wissamantoun's picture
added language generation
c59ebda
raw history blame
No virus
214 Bytes
#%%
from transformers import GPT2Tokenizer
# %%
tok = GPT2Tokenizer.from_pretrained("D:/ML/Models/aragpt2-medium", use_fast=False)
# %%
tok.pad_token = tok.eos_token
#%%
tok.pad_token_id = [tok.eos_token_id]
# %%