diff --git a/docs/source/en/api/models/hidream_image_transformer.md b/docs/source/en/api/models/hidream_image_transformer.md index 238d3007bb..5dbf40b5a1 100644 --- a/docs/source/en/api/models/hidream_image_transformer.md +++ b/docs/source/en/api/models/hidream_image_transformer.md @@ -37,24 +37,6 @@ transformer = HiDreamImageTransformer2DModel.from_single_file( ) ``` -If you are trying to use a GGUF checkpoint for the `HiDream-ai/HiDream-E1-Full` model, you will have to pass in a `config` argument to properly configure the model. This is because the HiDream I1 and E1 models share the same state dict keys, so it is currently not possible to automatically infer the model type from the checkpoint itself. - -```python -import torch -from diffusers import GGUFQuantizationConfig, HiDreamImageTransformer2DModel - -ckpt_path = "https://huggingface.co/ND911/HiDream_e1_full_bf16-ggufs/blob/main/hidream_e1_full_bf16-Q2_K.gguf" - -transformer = HiDreamImageTransformer2DModel.from_single_file( - ckpt_path, - quantization_config=GGUFQuantizationConfig(compute_dtype=torch.bfloat16), - config="HiDream-ai/HiDream-E1-Full", - subfolder="transformer", - torch_dtype=torch.bfloat16 -) -``` - - ## HiDreamImageTransformer2DModel [[autodoc]] HiDreamImageTransformer2DModel