Views
No views yet
translate [Source Language] to [Target Language]: 1from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
2from peft import PeftModel
3
4base_model_id = "google/mt5-base"
5peft_model_id = "achulz/mayan-mt5-qeqchi-mtl-adapter"
6
7tokenizer = AutoTokenizer.from_pretrained(peft_model_id)
8base_model = AutoModelForSeq2SeqLM.from_pretrained(base_model_id)
9model = PeftModel.from_pretrained(base_model, peft_model_id)
10
11# Example Translation
12prompt = "translate English to Q'eqchi': The dog is sleeping in the house."
13inputs = tokenizer(prompt, return_tensors="pt", max_length=128, truncation=True)
14outputs = model.generate(**inputs, max_new_tokens=128)
15print(tokenizer.decode(outputs[0], skip_special_tokens=True))tag POS Q'eqchi': 1# Example POS Tagging
2prompt = "tag POS Q'eqchi': Li Ch'ina'us naxk'e rib'..."
3inputs = tokenizer(prompt, return_tensors="pt", max_length=128, truncation=True)
4outputs = model.generate(**inputs, max_new_tokens=128)
5print(tokenizer.decode(outputs[0], skip_special_tokens=True))
6# Output format: Token (UD_TAG) Token (UD_TAG) ...tag semantic Q'eqchi': 1# Example Semantic Tagging
2prompt = "tag semantic Q'eqchi': Laj Carlos ..."
3inputs = tokenizer(prompt, return_tensors="pt", max_length=128, truncation=True)
4outputs = model.generate(**inputs, max_new_tokens=128)
5print(tokenizer.decode(outputs[0], skip_special_tokens=True))
6# Output format: Token (BIO-Tag) ...