nodes/smthemex-ComfyUI_MagiHuman
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update t5_gemma_model.py

Browse Source
This commit is contained in:
smthemex
2026-03-30 14:31:28 +08:00
committed by GitHub
parent 75952729eb
commit 8d8d81e6cd
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
inference/model/t5_gemma/t5_gemma_model.py
+1 -1
View File
@@ -7,7 +7,7 @@ from transformers import AutoTokenizer
try :
from transformers.models.t5gemma.modeling_t5gemma import T5GemmaEncoderModel,T5GemmaConfig
except:
from transformers.models.t5gemma import T5GemmaEncoderModel
from transformers.models.t5gemma import T5GemmaEncoderModel,T5GemmaConfig
from transformers import AutoModel, AutoTokenizer,AutoConfig
from safetensors.torch import load_file
from ...common import CPUOffloadWrapper, get_arch_memory