Update README.md

README.md

@@ -8,9 +8,12 @@ tags:
 ---
 
 
-This repo contains bitsandbytes 4bit-NF4 model weights for [OmniGen-v1](https://huggingface.co/Shitao/OmniGen-v1).
+This repo contains bitsandbytes 4bit-NF4 model weights for [OmniGen-v1](https://huggingface.co/Shitao/OmniGen-v1). For info about OmniGen see the [original model card](https://huggingface.co/Shitao/OmniGen-v1).
+
+
+- 8-bit weights: [gryan/OmniGen-v1-bnb-8bit](https://huggingface.co/gryan/OmniGen-v1-bnb-8bit)
+
 
-See the [original model card](https://huggingface.co/Shitao/OmniGen-v1) for more info.
 
 ## Usage
 Set up your environment by following the original [Quick Start Guide](https://huggingface.co/Shitao/OmniGen-v1#5-quick-start) before getting started.
@@ -22,7 +25,7 @@ Set up your environment by following the original [Quick Start Guide](https://hu
 from OmniGen import OmniGenPipeline, OmniGen
 
 # pass the quantized model in the pipeline
-model = OmniGen.from_pretrained('gryan/OmniGen-v1-bnb-nf4')
+model = OmniGen.from_pretrained('gryan/OmniGen-v1-bnb-4bit')
 pipe = OmniGenPipeline.from_pretrained("Shitao/OmniGen-v1", model=model)
 
 # proceed as normal!