Views
No views yet
model-00001-of-0000X.safetensors → Main model weightsconfig.jsontokenizer.json / tokenizer_config.jsongeneration_config.jsonspecial_tokens_map.jsonmodel.safetensors.index.jsonsafetensors format — No pickle risk.1from transformers import AutoModelForCausalLM, AutoTokenizer
2import torch
3
4model_name = "Shrijanagain/TIGER-OM"
5
6tokenizer = AutoTokenizer.from_pretrained(model_name)
7model = AutoModelForCausalLM.from_pretrained(
8 model_name,
9 torch_dtype=torch.bfloat16,
10 device_map="auto",
11 trust_remote_code=True
12)
13
14prompt = """You are SKT-OM, an advanced agentic AI with Think Mode enabled.
15User Query: Calculate training cost comparison and suggest best option..."""
16
17inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
18
19outputs = model.generate(
20 **inputs,
21 max_new_tokens=1024,
22 temperature=0.7,
23 top_p=0.9,
24 do_sample=True,
25 repetition_penalty=1.1
26)
27
28print(tokenizer.decode(outputs[0], skip_special_tokens=True))