Views
No views yet
transformers kütüphanesi ile çekebilirsiniz:1from transformers import AutoTokenizer, AutoModelForCausalLM
2import torch
3
4model_id = "OnurDemircioglu/OmniGPT-355M-Instruct"
5
6tokenizer = AutoTokenizer.from_pretrained(model_id)
7model = AutoModelForCausalLM.from_pretrained(
8 model_id,
9 torch_dtype=torch.float16,
10 attn_implementation="eager" # Windows CUDA cihazlar için önerilir
11).to("cuda")
12
13# Alpaca Prompt Formatı
14prompt = """Below is an instruction that describes a task. Write a response that appropriately completes the request.
15
16### Instruction:
17Explain what artificial intelligence is in simple terms.
18
19### Response:
20"""
21
22inputs = tokenizer(prompt, return_tensors="pt", truncation=True, max_length=150).to("cuda")
23
24output_ids = model.generate(
25 inputs["input_ids"],
26 attention_mask=inputs["attention_mask"],
27 max_new_tokens=60,
28 do_sample=True,
29 temperature=0.7,
30 top_p=0.9,
31 repetition_penalty=1.3,
32 pad_token_id=tokenizer.eos_token_id
33)
34
35response = tokenizer.decode(output_ids[0], skip_special_tokens=True)
36print(response)CUDA device-side assert hatası alınabilir. Giriş metnini her zaman max_length=150 gibi sınırlarla (truncation_side="left") koruma altına alın.repetition_penalty=1.3 veya üzeri değerlerin kullanılması zorunludur.