Update README.md

README.md

@@ -8,7 +8,7 @@ widget:
 library_name: transformers
 ---
 
-This model is randomly initialized, using the config from [https://huggingface.co/google/gemma-7b] but with smaller size. 
+This model is randomly initialized, using the config from [https://huggingface.co/google/gemma-7b-it] but with smaller size. 
 Note the model is in float16.
 
 Codes:
@@ -19,7 +19,7 @@ import torch
 import transformers
 import os
 
-model_id = 'google/gemma-7b'
+model_id = 'google/gemma-7b-it'
 save_path = '/tmp/yujiepan/gemma-tiny-random'
 repo_id = 'yujiepan/gemma-tiny-random'