diff --git a/python/llm/example/GPU/HF-Transformers-AutoModels/Model/gemma/generate.py b/python/llm/example/GPU/HF-Transformers-AutoModels/Model/gemma/generate.py index bbe4f68b..2ca0ab90 100644 --- a/python/llm/example/GPU/HF-Transformers-AutoModels/Model/gemma/generate.py +++ b/python/llm/example/GPU/HF-Transformers-AutoModels/Model/gemma/generate.py @@ -48,6 +48,7 @@ if __name__ == '__main__': load_in_4bit=True, optimize_model=True, trust_remote_code=True, + mixed_precision=True, use_cache=True) model = model.to('xpu')