Enable 2x faster inference with unsloth
This commit is contained in:
@@ -44,6 +44,9 @@ class QwQLlmEngine(LlmEngine):
|
||||
tokenizer = self._tokenizer,
|
||||
)
|
||||
|
||||
# enable unsloth optimizations
|
||||
FastLanguageModel.for_inference(self._model)
|
||||
|
||||
# Create inference pipeline
|
||||
self._pipeline = pipeline(
|
||||
"text-generation",
|
||||
|
||||
Reference in New Issue
Block a user