Update README.md

README.md

@@ -73,20 +73,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
 ```
 
 Then go to [app.thestage.ai](https://app.thestage.ai), login and generate API token from your profile page. Set up API token as follows: