Update README.md

README.md

@@ -12,7 +12,7 @@ tags:
 
 ## Model Description
 
-The **AMD Llama 350M Upgraded** is a transformer-based causal language model built on the Llama architecture, designed to generate human-like text. This model has been upgraded from the original AMD Llama model to provide enhanced performance with an increased parameter count of 500 million. It is suitable for various natural language processing tasks, including text generation, completion, and conversational applications.
+The **AMD Llama 350M Upgraded** is a transformer-based causal language model built on the Llama architecture, designed to generate human-like text. This model has been upgraded from the original AMD Llama 135M model to provide enhanced performance with an increased parameter count of 320 million. It is suitable for various natural language processing tasks, including text generation, completion, and conversational applications.
 
 ## Model Details
 
@@ -24,7 +24,7 @@ The **AMD Llama 350M Upgraded** is a transformer-based causal language model bui
 
 ## Usage
 
-To use the AMD Llama 500M Upgraded model, you can utilize the `transformers` library. Here’s a sample code snippet to get started:
+To use the AMD Llama 350M Upgraded model, you can utilize the `transformers` library. Here’s a sample code snippet to get started:
 
 ```python
 import torch