Spaces:
Runtime error
Runtime error
v5.1: revert INT8 — too lossy for 0.6B, keep LoRA merge + float32
Browse files
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", "
|
| 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 |
}
|