Merge pull request #1392 from bghira/bugfix/hidream-seq-len

bghira · web-flow · commit ae237ccb5548 · 2025-04-14T19:35:53.000-06:00
add missing seq len
diff --git a/helpers/models/hidream/model.py b/helpers/models/hidream/model.py
@@ -308,6 +308,7 @@ def check_user_config(self):
             raise ValueError(
                 f"{self.NAME} does not support fp8-quanto. Please use fp8-torchao or int8 precision level instead."
             )
+        t5_max_length = 128
         if self.config.tokenizer_max_length is None or self.config.tokenizer_max_length == 0:
             logger.warning(
                 f"Setting T5 XXL tokeniser max length to {t5_max_length} for {self.NAME}."

Original file line number	Diff line number	Diff line change
`@@ -308,6 +308,7 @@ def check_user_config(self):`
`308`	`308`	`raise ValueError(`
`309`	`309`	`f"{self.NAME} does not support fp8-quanto. Please use fp8-torchao or int8 precision level instead."`
`310`	`310`	`)`
	`311`	`+ t5_max_length = 128`
`311`	`312`	`if self.config.tokenizer_max_length is None or self.config.tokenizer_max_length == 0:`
`312`	`313`	`logger.warning(`
`313`	`314`	`f"Setting T5 XXL tokeniser max length to {t5_max_length} for {self.NAME}."`