Views
No views yet
1max_seq_length = 256
2dtype = None
3load_in_4bit = True
4
5alpaca_prompt = """
6أدناه تعليمة تصف مهمة مقترنة بمدخلات تضيف سياق إن وجدت. اكتب إجابة تتناسب مع التعليمة والمدخلات مع الحفاظ على القيم واﻵداب العامة.
7
8### التعليمة:
9{}
10
11### المدخلات:
12{}
13
14### اﻹجابة:
15{}"""
16
17from unsloth import FastLanguageModel
18
19model, tokenizer = FastLanguageModel.from_pretrained(
20 model_name = "akhooli/llama31ft",
21 max_seq_length = max_seq_length,
22 dtype = dtype,
23 load_in_4bit = load_in_4bit,
24 )
25model = FastLanguageModel.for_inference(model)
26
27inputs = tokenizer(
28[
29 alpaca_prompt.format(
30 "اكتب قصيدة شعرية قصيرة", # instruction
31 "بحر البسيط", # input
32 "", # output - leave this blank for generation!
33 )
34], return_tensors = "pt").to("cuda")
35
36outputs = model.generate(**inputs, max_new_tokens = 256, use_cache = True,temperature=0.95)
37r = tokenizer.batch_decode(outputs)
38from pprint import pprint
39pprint(r)