IMvision12 commited on
Commit
487192b
·
verified ·
1 Parent(s): 9e67675

Rename model_class to the split generation class

Browse files
Files changed (1) hide show
  1. kf_config.json +2 -2
kf_config.json CHANGED
@@ -2,7 +2,7 @@
2
  "library_name": "kerasformers",
3
  "kerasformers_version": "1.2.1",
4
  "model_module": "kerasformers.models.gemma4",
5
- "model_class": "Gemma4Generate",
6
  "variant": "gemma-4-e2b-it",
7
  "weights": "model.weights.json",
8
  "schema_version": 2,
@@ -112,4 +112,4 @@
112
  "audio_token_id": 258881,
113
  "pad_token_id": 0,
114
  "use_bidirectional_vision": false
115
- }
 
2
  "library_name": "kerasformers",
3
  "kerasformers_version": "1.2.1",
4
  "model_module": "kerasformers.models.gemma4",
5
+ "model_class": "Gemma4ConditionalGenerate",
6
  "variant": "gemma-4-e2b-it",
7
  "weights": "model.weights.json",
8
  "schema_version": 2,
 
112
  "audio_token_id": 258881,
113
  "pad_token_id": 0,
114
  "use_bidirectional_vision": false
115
+ }