Update README.md

README.md

@@ -38,7 +38,7 @@ __Goals of Elastic Models:__
 
 ## Inference
 
-Currently, our demo model only supports 1024x1024 outputs without batching. This will be updated in the near future.
+Currently, our demo model only supports 1024x1024, 768x768 and 512x512 outputs without batching. This will be updated in the near future.
 To infer our models, you just need to replace `diffusers` import with `elastic_models.diffusers`:
 
 ```python
@@ -87,6 +87,9 @@ pip install elastic_models[blackwell]\
  --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 -U --pre torch --index-url https://download.pytorch.org/whl/nightly/cu128
+pip install -U --pre torchvision --index-url https://download.pytorch.org/whl/nightly/cu128
+
 
 pip install flash_attn==2.7.3 --no-build-isolation
 pip uninstall apex