mirror of
https://github.com/vladmandic/automatic
synced 2026-09-19 17:24:32 +02:00
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:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user