Views
No views yet
mt5-small adaptada al català. Ha estat entrenada amb la tècnica de span masking sobre 1.500.000 frases correctes en català per millorar la comprensió de la llengua i afinar l’ús semàntic i sintàctic en català.mt5-small original en contextos catalansgoogle/mt5-smallTextInfilling)1from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
2
3model = AutoModelForSeq2SeqLM.from_pretrained("Oriolshhh/mt5-small-ca-spanmask-1.5m")
4tokenizer = AutoTokenizer.from_pretrained("Oriolshhh/mt5-small-ca-spanmask-1.5m")
5
6text = "Avui <extra_id_0> una bonica <extra_id_1> per sortir."
7inputs = tokenizer(text, return_tensors="pt")
8outputs = model.generate(**inputs)
9print(tokenizer.decode(outputs[0], skip_special_tokens=True))