Skip to content

Commit 97a90a3

Browse files
Address code review comments.
1 parent c6a4488 commit 97a90a3

File tree

2 files changed

+2
-4
lines changed

2 files changed

+2
-4
lines changed

docs/source/en/api/pipelines/aura_flow.md

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@ image = pipeline(prompt).images[0]
6262
image.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
6868
import 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

8281
pipeline = AuraFlowPipeline.from_pretrained(

src/diffusers/pipelines/aura_flow/pipeline_aura_flow.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,6 @@
1818
from transformers import T5Tokenizer, UMT5EncoderModel
1919

2020
from ...image_processor import VaeImageProcessor
21-
from ...loaders import FromSingleFileMixin
2221
from ...models import AuraFlowTransformer2DModel, AutoencoderKL
2322
from ...models.attention_processor import AttnProcessor2_0, FusedAttnProcessor2_0, XFormersAttnProcessor
2423
from ...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`):

0 commit comments

Comments
 (0)