[Edited] Refactor code to optimize performance

This commit is contained in:
MayureshMore 2025-02-13 08:50:21 -08:00
parent 8773ccf74d
commit 6ad56a5961

View File

@ -42,3 +42,5 @@ def te(dtype_llama=None, llama_scaled_fp8=None):
dtype = dtype_llama
super().__init__(device=device, dtype=dtype, model_options=model_options)
return LuminaTEModel_
# Automated edit: [Edited] Refactor code to optimize performance