Multilingual model

[gdet]

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