fix(example): set model_type to load for gemma3 text (#3242)

* fix: set model_type to load for gemma3 text * chore: simplify * chore: unify
2025-11-04 07:35:07 +07:00
parent ed58fa8a75
commit 26f05b6008
3 changed files with 9 additions and 6 deletions
--- a/examples/gemma3/gemma-3-4b-qlora.yml
+++ b/examples/gemma3/gemma-3-4b-qlora.yml
@@ -1,5 +1,8 @@
 base_model: google/gemma-3-4b-it

+# Need to set else transformers tries to load vision too
+model_type: Gemma3ForCausalLM
+
 load_in_4bit: true

 # gemma3 doesn't seem to play nice with ddp