arthu1 commited on
Commit
26344b5
·
1 Parent(s): ca997b4

v5.1: revert INT8 — too lossy for 0.6B, keep LoRA merge + float32

Browse files
Files changed (1) hide show
  1. app.py +2 -13
app.py CHANGED
@@ -80,18 +80,7 @@ def _load_model():
80
  )
81
 
82
  MODEL.eval()
83
-
84
- # Dynamic INT8 quantization — now safe since LoRA is merged
85
- try:
86
- MODEL = torch.quantization.quantize_dynamic(
87
- MODEL, {torch.nn.Linear}, dtype=torch.qint8,
88
- )
89
- INFERENCE_MODE = "pytorch-int8"
90
- print("INT8 quantization applied")
91
- except Exception as e:
92
- INFERENCE_MODE = "pytorch"
93
- print(f"Quantization skipped: {e}")
94
-
95
  print(f"Model ready: {INFERENCE_MODE}")
96
 
97
  except Exception as e:
@@ -110,7 +99,7 @@ def health():
110
  "version": "5.0.0",
111
  "architecture": "Qwen3-0.6B + LoRA r=64 (merged)",
112
  "inference": INFERENCE_MODE,
113
- "features": ["streaming", "thinking", "int8"],
114
  "model_dir": MODEL_DIR,
115
  "error": LOAD_ERROR,
116
  }
 
80
  )
81
 
82
  MODEL.eval()
83
+ INFERENCE_MODE = "pytorch-merged"
 
 
 
 
 
 
 
 
 
 
 
84
  print(f"Model ready: {INFERENCE_MODE}")
85
 
86
  except Exception as e:
 
99
  "version": "5.0.0",
100
  "architecture": "Qwen3-0.6B + LoRA r=64 (merged)",
101
  "inference": INFERENCE_MODE,
102
+ "features": ["streaming", "thinking", "merged"],
103
  "model_dir": MODEL_DIR,
104
  "error": LOAD_ERROR,
105
  }