Files changed (1) hide show
  1. README.md +3 -1
README.md CHANGED
@@ -5,7 +5,9 @@ tags:
5
  - flax
6
  - text-generation
7
  - transformers
8
- - google/gemma-2b # Add the specific model name as a tag
 
 
9
  ---
10
 
11
  # google/gemma-2b - JAX/Flax
 
5
  - flax
6
  - text-generation
7
  - transformers
8
+ - google/gemma-2b
9
+ base_model:
10
+ - google/gemma-2b
11
  ---
12
 
13
  # google/gemma-2b - JAX/Flax