Views
No views yet

1from transformers import AutoTokenizer, AutoModelForCausalLM
2import torch
3
4model_name = "Hyggshi-AI/HOS-OSS-3.08B"
5
6tokenizer = AutoTokenizer.from_pretrained(model_name)
7model = AutoModelForCausalLM.from_pretrained(model_name)
8
9prompt = "User: Write a Python Hello World
10Assistant:"
11
12inputs = tokenizer(prompt, return_tensors="pt")
13
14with torch.no_grad():
15 outputs = model.generate(
16 **inputs,
17 max_new_tokens=512,
18 temperature=0.7
19 )
20
21print(tokenizer.decode(outputs[0], skip_special_tokens=True))