Views
No views yet
0.3878 - 0.4068 at step 2000!1import torch
2from transformers import AutoTokenizer, AutoModelForCausalLM
3
4REPO_ID = "sana0756/hipponet-gemma-398moe-omega2"
5
6tokenizer = AutoTokenizer.from_pretrained(REPO_ID)
7model = AutoModelForCausalLM.from_pretrained(
8 REPO_ID,
9 torch_dtype=torch.bfloat16,
10 device_map="auto"
11)
12
13prompt = """<start_of_turn>user
14Aşağıdaki Python fonksiyonunu analiz et ve optimize edilmiş halini açıkla:
15def fibonacci(n):
16 if n <= 1: return n
17 return fibonacci(n-1) + fibonacci(n-2)<end_of_turn>
18<start_of_turn>model
19"""
20
21inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
22outputs = model.generate(**inputs, max_new_tokens=512, temperature=0.7, top_p=0.9)
23print(tokenizer.decode(outputs[0], skip_special_tokens=True))