Ukrainian-to-English translation model based on Yehor/kulyk-uk-en (LFM2-350M), improved with GRPO using calibrated guardrail rewards on WikiMatrix data.
fromtransformersimportAutoModelForCausalLM,AutoTokenizermodel=AutoModelForCausalLM.from_pretrained("iamthewalrus67/kulyk-uk-en-grpo",trust_remote_code=True)tokenizer=AutoTokenizer.from_pretrained("iamthewalrus67/kulyk-uk-en-grpo",trust_remote_code=True)prompt="Translate the text to English:\nПогода сьогодні чудова."input_ids=tokenizer.apply_chat_template([{"role":"user","content":prompt}],add_generation_prompt=True,return_tensors="pt",tokenize=True).to(model.device)output=model.generate(input_ids,max_new_tokens=256,do_sample=False,repetition_penalty=1.05)print(tokenizer.decode(output[0][input_ids.shape[1]:],skip_special_tokens=True))