More work on compiling models

This commit is contained in:
Jaret Burkett
2026-03-31 12:11:56 -06:00
parent 358d684f6f
commit 4a3251640a
2 changed files with 8 additions and 2 deletions

View File

@@ -695,6 +695,10 @@ class ModelConfig:
# compile the model with torch compile
self.compile = kwargs.get("compile", False)
if self.compile and self.quantize:
print("Warning: You cannot compile a quantized model. Disabling compile.")
self.compile = False
# kwargs to pass to the model
self.model_kwargs = kwargs.get("model_kwargs", {})