Update README.md

README.md

@@ -25,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-fp16-bnb-4bit')
+model = OmniGen.from_pretrained('gryan/OmniGen-v1-fp16-bnb-4bit', dtype=torch.float16)
 pipe = OmniGenPipeline.from_pretrained("Shitao/OmniGen-v1", model=model)
 
 # proceed as normal!