madtune commited on
Commit
2e87f45
·
verified ·
1 Parent(s): fa8167b

Upload model_index.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. model_index.json +2 -2
model_index.json CHANGED
@@ -3,6 +3,6 @@
3
  "_diffusers_version": "0.31.0",
4
  "transformer": ["diffusers", "PixelDiTModel"],
5
  "scheduler": ["diffusers", "FlowMatchEulerDiscreteScheduler"],
6
- "text_encoder": ["transformers", "AutoModelForCausalLM"],
7
- "tokenizer": ["transformers", "AutoTokenizer"]
8
  }
 
3
  "_diffusers_version": "0.31.0",
4
  "transformer": ["diffusers", "PixelDiTModel"],
5
  "scheduler": ["diffusers", "FlowMatchEulerDiscreteScheduler"],
6
+ "text_encoder": ["transformers", "PreTrainedModel"],
7
+ "tokenizer": ["transformers", "PreTrainedTokenizerFast"]
8
  }