Update README.md
Browse files
README.md
CHANGED
|
@@ -12,13 +12,19 @@ File size:
|
|
| 12 |
- ~335MB ae.safetensors (VAE, full architecture — hardcoded in vllm-omni)
|
| 13 |
- ~41MB ema.safetensors (LLM + ViT + connectors, 1 layer each)
|
| 14 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 15 |
### Example usage:
|
| 16 |
|
| 17 |
```python
|
| 18 |
from vllm_omni.entrypoints.omni import Omni
|
| 19 |
|
| 20 |
omni = Omni(
|
| 21 |
-
model="zhengyuansu/bagel-tiny-random",
|
|
|
|
| 22 |
stage_configs_path="path/to/bagel_sharedmemory_2gpu_ci.yaml",
|
| 23 |
custom_pipeline_args={
|
| 24 |
"pipeline_class": "examples.flowgrpo_trainer.vllm_omni.pipeline_bagel.BagelPipelineWithLogProb"
|
|
|
|
| 12 |
- ~335MB ae.safetensors (VAE, full architecture — hardcoded in vllm-omni)
|
| 13 |
- ~41MB ema.safetensors (LLM + ViT + connectors, 1 layer each)
|
| 14 |
|
| 15 |
+
## Acknowledgements
|
| 16 |
+
|
| 17 |
+
🎉 Special thanks to [@zhengyuansu](https://huggingface.co/zhengyuansu) for contributing this model!
|
| 18 |
+
|
| 19 |
+
|
| 20 |
### Example usage:
|
| 21 |
|
| 22 |
```python
|
| 23 |
from vllm_omni.entrypoints.omni import Omni
|
| 24 |
|
| 25 |
omni = Omni(
|
| 26 |
+
# model="zhengyuansu/bagel-tiny-random",
|
| 27 |
+
model="tiny-random/bagel",
|
| 28 |
stage_configs_path="path/to/bagel_sharedmemory_2gpu_ci.yaml",
|
| 29 |
custom_pipeline_args={
|
| 30 |
"pipeline_class": "examples.flowgrpo_trainer.vllm_omni.pipeline_bagel.BagelPipelineWithLogProb"
|