Views
No views yet
transformers remote code.1from transformers import AutoTokenizer, AutoModelForCausalLM
2import torch
3
4repo = "himalaya-ai/himalayagpt-0.5b"
5tok = AutoTokenizer.from_pretrained(repo, trust_remote_code=True)
6model = AutoModelForCausalLM.from_pretrained(
7 repo,
8 trust_remote_code=True,
9 torch_dtype=torch.bfloat16 if torch.cuda.is_available() else torch.float32,
10 device_map="auto" if torch.cuda.is_available() else None,
11)
12
13prompt = "नेपालको राजधानी "
14ids = tok(prompt, return_tensors="pt").input_ids.to(model.device)
15out = model.generate(ids, max_new_tokens=64)
16print(tok.decode(out[0], skip_special_tokens=True))trust_remote_code=True).based15_harl_fulltokens_sdpa_bs32133632