Update README.md

README.md

@@ -94,19 +94,13 @@ To work with our models just run these lines in your terminal:
 
 ```shell
 pip install thestage
-pip install elastic_models==0.0.3\
+pip install elastic_models[nvidia]\
  --index-url https://thestage.jfrog.io/artifactory/api/pypi/pypi-thestage-ai-production/simple\
  --extra-index-url https://pypi.nvidia.com\
  --extra-index-url https://pypi.org/simple
 
 pip install flash_attn==2.7.3 --no-build-isolation
 pip uninstall apex
-echo "{
-    "meta-llama/Llama-3.2-1B-Instruct": 6,
-    "mistralai/Mistral-7B-Instruct-v0.3": 7,
-    "black-forest-labs/FLUX.1-schnell": 1,
-    "black-forest-labs/FLUX.1-dev": 5
-}" > model_name_id.json
 export ELASTIC_MODEL_ID_MAPPING=./model_name_id.json
 ```