Shitao

/

OmniGen-v1

like 228

Hi, Thanks for the wonderful model,
I'm trying to quantize the model but it seems like model_index.json is missing:

pipe = DiffusionPipeline.from_pretrained("Shitao/OmniGen-v1", torch_dtype=torch.float16).to("cuda")

I got:

Entry Not Found for url: https://huggingface.co./Shitao/OmniGen-v1/resolve/main/model_index.json.

Can you please add it? According to the documentation,

The model_index.json follows the example format:

{
  "_class_name": "StableDiffusionPipeline",
  "_diffusers_version": "0.6.0",
  "feature_extractor": [
    "transformers",
    "CLIPImageProcessor"
  ],
  "safety_checker": [
    "stable_diffusion",
    "StableDiffusionSafetyChecker"
  ],
  "scheduler": [
    "diffusers",
    "PNDMScheduler"
  ],
  "text_encoder": [
    "transformers",
    "CLIPTextModel"
  ],
  "tokenizer": [
    "transformers",
    "CLIPTokenizer"
  ],
  "unet": [
    "diffusers",
    "UNet2DConditionModel"
  ],
  "vae": [
    "diffusers",
    "AutoencoderKL"
  ]
}