README.md

README.md

@@ -52,11 +52,22 @@ STM32F4 serisi için HAL kütüphanesi kullanan gömülü sistem kodu üretir.
 ```python
 from transformers import AutoTokenizer, AutoModelForCausalLM
 from peft import PeftModel
+import torch
+
+print("🔄 Model yükleniyor...")
+
+base_model = AutoModelForCausalLM.from_pretrained(
+    "bigcode/starcoder2-3b",
+    torch_dtype=torch.float16,  
+    device_map="auto"
+)
 
-base_model = AutoModelForCausalLM.from_pretrained("bigcode/starcoder2-3b")
 model = PeftModel.from_pretrained(base_model, "MuratKomurcu/starcoder2-stm32-turkish")
 tokenizer = AutoTokenizer.from_pretrained("MuratKomurcu/starcoder2-stm32-turkish")
 
+print("✅ Model hazır!")
+
+# TEST
 prompt = '''### Instruction:
 Write complete STM32F401RE LED blink code with main function
 
@@ -65,10 +76,6 @@ LED on PA5, blink every 1000ms, include SystemClock_Config
 
 ### Response:
 '''
-
-inputs = tokenizer(prompt, return_tensors="pt")
-outputs = model.generate(**inputs, max_new_tokens=2048, temperature=0.1)
-print(tokenizer.decode(outputs[0]))
 ```
 
 ## Eğitim Detayları