add dep

kylesayrs · kylesayrs · commit 9dc45e661782 · 2025-10-14T00:46:02.000-04:00
Signed-off-by: Kyle Sayers &lt;kylesayrs@gmail.com&gt;
diff --git a/tests/llmcompressor/transformers/tracing/test_models.py b/tests/llmcompressor/transformers/tracing/test_models.py
@@ -50,7 +50,7 @@
             "text",
             [],
         ),
-        ("google/gemma-3n-E2B-it", AutoModelForCausalLM, None, "text", []),
+        ("google/gemma-3n-E2B-it", AutoModelForCausalLM, None, "text", ["timm"]),
         ("unsloth/DeepSeek-R1-0528-BF16", AutoModelForCausalLM, None, "text", []),
         # --- vision ---
         (

Original file line number	Diff line number	Diff line change
`@@ -50,7 +50,7 @@`
`50`	`50`	`"text",`
`51`	`51`	`[],`
`52`	`52`	`),`
`53`		`- ("google/gemma-3n-E2B-it", AutoModelForCausalLM, None, "text", []),`
	`53`	`+ ("google/gemma-3n-E2B-it", AutoModelForCausalLM, None, "text", ["timm"]),`
`54`	`54`	`("unsloth/DeepSeek-R1-0528-BF16", AutoModelForCausalLM, None, "text", []),`
`55`	`55`	`# --- vision ---`
`56`	`56`	`(`