from transformers import AutoModelForCausalLM, AutoTokenizer #تحميل النموذجmodel_id = "Etherll/Tashkeel-350M-v2"model = AutoModelForCausalLM.from_pretrained( model_id, device_map="auto", torch_dtype="bfloat16",)tokenizer = AutoTokenizer.from_pretrained(model_id) # إضافة التشكيلprompt = "السلام عليكم" input_ids = tokenizer.apply_chat_template( [{"role": "user", "content": "قم بتشكيل هذا النص "+ ":\n"+prompt}], add_generation_prompt=True, return_tensors="pt", tokenize=True,).to(model.device) output = model.generate( input_ids, do_sample=False, ) print(tokenizer.decode(output[0, input_ids.shape[-1]:], skip_special_tokens=True))