huggingface / transfer-learning-conv-ai

🦄 State-of-the-Art Conversational AI with Transfer Learning
MIT License
1.73k stars 431 forks source link

Multilingual model #57

Open gdet opened 4 years ago

gdet commented 4 years ago

Hello, sorry if this is a silly question. Can I somehow use the multilingual model with this code? A changed the tokenized and model to

tokenizer = AutoTokenizer.from_pretrained("bert-base-multilingual-cased")
model = AutoModelWithLMHead.from_pretrained("bert-base-multilingual-cased")

but then I don't understand how to change functions def update() and def inference() which use the loss function.

If someone could explain or give me some basic steps it will be very helpful. Thank you