Views
No views yet
Transformers both for GPU and CPU enabled envs:1from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
2import torch
3
4model_name = "minilingua-ai/MiniLingua-1b-Instruct"
5
6tokenizer = AutoTokenizer.from_pretrained(model_name)
7model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto", dtype=torch.float16)
8gen = pipeline("text-generation", model=model, tokenizer=tokenizer, trust_remote_code=True)
9
10prompt = "Translate from Bulgarian: Здравейте! Как сте? Translation:"
11out = gen(prompt, max_new_tokens=128, do_sample=False)
12print(out[0])