Views
No views yet
1from transformers import AutoTokenizer, AutoModelForCausalLM
2from peft import PeftModel
3import torch
4
5print("🔄 Model yükleniyor...")
6
7base_model = AutoModelForCausalLM.from_pretrained(
8 "bigcode/starcoder2-3b",
9 torch_dtype=torch.float16,
10 device_map="auto"
11)
12
13model = PeftModel.from_pretrained(base_model, "MuratKomurcu/starcoder2-stm32-turkish")
14tokenizer = AutoTokenizer.from_pretrained("MuratKomurcu/starcoder2-stm32-turkish")
15
16print("✅ Model hazır!")
17
18# TEST
19prompt = '''### Instruction:
20Write complete STM32F401RE LED blink code with main function
21
22### Input:
23LED on PA5, blink every 1000ms, include SystemClock_Config
24
25### Response:
26'''1@misc{starcoder2-stm32-turkish,
2 author = {Murat},
3 title = {STM32 Code Generator - Turkish Fine-tuned},
4 year = {2024},
5 publisher = {HuggingFace},
6 howpublished = {\url{https://huggingface.co/MuratKomurcu/starcoder2-stm32-turkish}}
7}