add google gemma-4-e2b and gemma-4-e4b support

- add to prompt enhance models, img2img, and models_cls
- add to vlm/caption models
- update _load_gemma class selection for gemma 4
- bump transformers pin to 5.5.0.dev for gemma4 architecture support
This commit is contained in:
CalamitousFelicitousness
2026-04-09 03:21:50 +01:00
parent 13fcbd1112
commit 0e5d351099
4 changed files with 12 additions and 3 deletions
+5
View File
@@ -109,6 +109,8 @@ class Options:
'google/gemma-3-4b-it',
'google/gemma-3n-E2B-it',
'google/gemma-3n-E4B-it',
'google/gemma-4-E2B-it',
'google/gemma-4-E4B-it',
# Gemma Finetunes
'nidum/Nidum-Gemma-3-4B-it-Uncensored',
'allura-org/Gemma-3-Glitter-4B',
@@ -173,6 +175,8 @@ class Options:
'google/gemma-3-4b-it': {},
'google/gemma-3n-E2B-it': {},
'google/gemma-3n-E4B-it': {},
'google/gemma-4-E2B-it': {},
'google/gemma-4-E4B-it': {},
# Gemma Finetunes
'nidum/Nidum-Gemma-3-4B-it-Uncensored': {},
'allura-org/Gemma-3-Glitter-4B': {},
@@ -267,6 +271,7 @@ class Options:
'qwen2_5_vl': 'Qwen2_5_VLForConditionalGeneration',
'qwen2_vl': 'Qwen2VLForConditionalGeneration',
'mistral3': 'Mistral3ForConditionalGeneration',
'gemma4': 'Gemma4ForConditionalGeneration',
}
# default = list(models)[1] # gemma-3-4b-it