From e195f2776b2ddb74271ffcc0faff5b84eb4459d2 Mon Sep 17 00:00:00 2001 From: Timothy McCallum Date: Thu, 16 Jan 2025 11:56:26 +1000 Subject: [PATCH] Add supported architectures note to readme file. This prevents users going to all the trouble of installing Cosmos, downloading models and then finding out that their architecture is not compatible. --- README.md | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/README.md b/README.md index e2cdbe82..6f9b9b34 100644 --- a/README.md +++ b/README.md @@ -34,6 +34,10 @@ Details of the platform is described in the [Cosmos paper](https://research.nvid | [Cosmos-1.0-Autoregressive-13B-Video2World](https://huggingface.co/nvidia/Cosmos-1.0-Autoregressive-13B-Video2World) | Video + Text based future visual world generation | [Inference](cosmos1/models/autoregressive/README.md) | | [Cosmos-1.0-Guardrail](https://huggingface.co/nvidia/Cosmos-1.0-Guardrail) | Guardrail contains pre-Guard and post-Guard for safe use | Embedded in model inference scripts | +## Supported Architectures + +Cosmos only supports GPUs based on the Ampere, Hopper, Ada Lovelace, and Blackwell architectures. Other architectures are not supported. + ## Example Usage ### Inference