EthanZyh commited on
Commit
0b94f8e
·
1 Parent(s): 97c7ba6

updated README

Browse files
Files changed (1) hide show
  1. README.md +19 -0
README.md CHANGED
@@ -1,3 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
 
2
  ![Cosmos Logo](https://github.com/NVIDIA/Cosmos/raw/main/assets/cosmos-logo.png)
3
 
 
1
+ ## How to Use
2
+
3
+ ```python
4
+ from transformers import AutoModel
5
+
6
+ model = AutoModel.from_pretrained(
7
+ "EthanZyh/DiffusionText2WorldGeneration",
8
+ cache_dir="./cache",
9
+ trust_remote_code=True,
10
+ # turn on offloading on a low GPU memory machine:
11
+ # offload_network=True,
12
+ # offload_tokenizer=True,
13
+ # offload_text_encoder_model=True,
14
+ # offload_prompt_upsampler=True,
15
+ # offload_guardrail_models=True,
16
+ )
17
+ prompt = "Some text prompt to generate a video"
18
+ model(prompt)
19
+ ```
20
 
21
  ![Cosmos Logo](https://github.com/NVIDIA/Cosmos/raw/main/assets/cosmos-logo.png)
22