Update README.md

README.md

@@ -51,7 +51,7 @@ Detailed efficiency metrics coming soon!
 
 You can run the smashed model with these steps:
 
-0. Check requirements from the original repo 22h/open-cabrita3b installed. In particular, check python, cuda, and transformers versions.
+0. Check requirements from the original repo google/gemma-1.1-2b-it installed. In particular, check python, cuda, and transformers versions.
 1. Make sure that you have installed quantization related packages.
     ```bash
     pip install transformers accelerate bitsandbytes>0.37.0
@@ -81,7 +81,7 @@ The configuration info are in `smash_config.json`.
 
 ## Credits & License
 
-The license of the smashed model follows the license of the original model. Please check the license of the original model 22h/open-cabrita3b before using this model which provided the base model. The license  of the `pruna-engine` is [here](https://pypi.org/project/pruna-engine/) on Pypi.
+The license of the smashed model follows the license of the original model. Please check the license of the original model google/gemma-1.1-2b-it before using this model which provided the base model. The license  of the `pruna-engine` is [here](https://pypi.org/project/pruna-engine/) on Pypi.
 
 ## Want to compress other models?