File tree Expand file tree Collapse file tree 2 files changed +2
-4
lines changed
docs/source/en/api/pipelines
src/diffusers/pipelines/aura_flow Expand file tree Collapse file tree 2 files changed +2
-4
lines changed Original file line number Diff line number Diff line change @@ -62,7 +62,7 @@ image = pipeline(prompt).images[0]
6262image.save(" auraflow.png" )
6363```
6464
65- Loading [ GGUF checkpoints] ( https://huggingface.co/city96/AuraFlow-v0.3- gguf ) are also supported:
65+ Loading [ GGUF checkpoints] ( https://huggingface.co/docs/diffusers/quantization/ gguf ) are also supported:
6666
6767``` py
6868import torch
@@ -76,7 +76,6 @@ transformer = AuraFlowTransformer2DModel.from_single_file(
7676 " https://huggingface.co/city96/AuraFlow-v0.3-gguf/blob/main/aura_flow_0.3-Q2_K.gguf" ,
7777 quantization_config = GGUFQuantizationConfig(compute_dtype = torch.bfloat16),
7878 torch_dtype = torch.bfloat16,
79- config = " fal/AuraFlow-v0.3" ,
8079)
8180
8281pipeline = AuraFlowPipeline.from_pretrained(
Original file line number Diff line number Diff line change 1818from transformers import T5Tokenizer , UMT5EncoderModel
1919
2020from ...image_processor import VaeImageProcessor
21- from ...loaders import FromSingleFileMixin
2221from ...models import AuraFlowTransformer2DModel , AutoencoderKL
2322from ...models .attention_processor import AttnProcessor2_0 , FusedAttnProcessor2_0 , XFormersAttnProcessor
2423from ...schedulers import FlowMatchEulerDiscreteScheduler
@@ -112,7 +111,7 @@ def retrieve_timesteps(
112111 return timesteps , num_inference_steps
113112
114113
115- class AuraFlowPipeline (DiffusionPipeline , FromSingleFileMixin ):
114+ class AuraFlowPipeline (DiffusionPipeline ):
116115 r"""
117116 Args:
118117 tokenizer (`T5TokenizerFast`):
You can’t perform that action at this time.
0 commit comments