nightknocker commited on
Commit
6872224
·
verified ·
1 Parent(s): c5502c4

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +9 -0
README.md CHANGED
@@ -16,6 +16,15 @@ It was trained on both T5 (text) and the [AnimaTextToImagePipeline](https://hugg
16
 
17
  *The project is a work in progress, see the image above.*
18
 
 
 
 
 
 
 
 
 
 
19
  ## References
20
 
21
  - [Recurrent Qwen](https://huggingface.co/nightknocker/recurrent-qwen3-z-image-turbo)
 
16
 
17
  *The project is a work in progress, see the image above.*
18
 
19
+ ## Inference
20
+
21
+ ```python
22
+ bert = CosmosBert.from_pretrained('nightknocker/cosmos-bert')
23
+ tokenizer = AutoTokenizer.from_pretrained('nightknocker/cosmos-bert')
24
+ inputs = tokenizer(text, return_tensors='pt').to('cuda')
25
+ crossattn_emb = bert.forward(**inputs, return_dict=True).last_hidden_state
26
+ ```
27
+
28
  ## References
29
 
30
  - [Recurrent Qwen](https://huggingface.co/nightknocker/recurrent-qwen3-z-image-turbo)