optimum-intel-internal-testing
/

tiny-random-gemma4-31B

Model card Files Files and versions

pnevskaiaan commited on May 11

Commit

737d788

·

verified ·

1 Parent(s): eb349c6

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -11,7 +11,7 @@ from transformers import (
 def generate_vlm_model(output_dir="./tiny-random-gemma4-31B"):
     from transformers import AutoConfig, AutoProcessor, AutoTokenizer, Gemma4ForConditionalGeneration
-    model_id = "google/gemma-4-31B"
     config = AutoConfig.from_pretrained(model_id)
     # Text config

 def generate_vlm_model(output_dir="./tiny-random-gemma4-31B"):
     from transformers import AutoConfig, AutoProcessor, AutoTokenizer, Gemma4ForConditionalGeneration
+    model_id = "google/gemma-4-31B-it"
     config = AutoConfig.from_pretrained(model_id)
     # Text config