More work on compiling models

This commit is contained in:
Jaret Burkett committed 2026-03-31 12:11:56 -06:00
1 parent 358d684f6f
commit 4a3251640a
2 files changed
+8 -2

No files matched your search

+4
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", {})