Update download_model.py

download_model.py

@@ -1,5 +1,4 @@
 from transformers import VisionEncoderDecoderModel, TrOCRProcessor
-import torch
 import os
 
 MODEL_ID = 'anuashok/ocr-captcha-v3'
@@ -11,15 +10,8 @@ os.makedirs(SAVE_PATH, exist_ok=True)
 processor = TrOCRProcessor.from_pretrained(MODEL_ID)
 model = VisionEncoderDecoderModel.from_pretrained(MODEL_ID)
 
-print("Quantizing to INT8...")
-quantized_model = torch.quantization.quantize_dynamic(
-    model,
-    {torch.nn.Linear},
-    dtype=torch.qint8
-)
-
-print("Saving quantized model to ./model...")
+print("Saving to ./model...")
 processor.save_pretrained(SAVE_PATH)
-quantized_model.save_pretrained(SAVE_PATH)
+model.save_pretrained(SAVE_PATH)
 
-print("✅ Quantized model saved successfully")
+print("✅ Model saved successfully")