More work on compiling models
This commit is contained in:
1 parent
358d684f6f
commit
4a3251640a
2 files changed
+8
-2
No files matched your search
@@ -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", {})
|
||||
|
||||
|
||||
Reference in new issue
Block a user