Instructions to use fal/AuraFlow with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use fal/AuraFlow with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("fal/AuraFlow", dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0] - Inference
- Notebooks
- Google Colab
- Kaggle
- Local Apps
- Draw Things
- DiffusionBee
Error using Model
I tried to use the model as described in the model card, and I'm encountering the following error:
File "C:\Python312\Lib\site-packages\huggingface_hub\utils_validators.py", line 114, in _inner_fn
return fn(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^
File "C:\Python312\Lib\site-packages\diffusers-0.30.0.dev0-py3.12.egg\diffusers\pipelines\pipeline_utils.py", line 881, in from_pretrained
loaded_sub_model = load_sub_model(
^^^^^^^^^^^^^^^
File "C:\Python312\Lib\site-packages\diffusers-0.30.0.dev0-py3.12.egg\diffusers\pipelines\pipeline_loading_utils.py", line 703, in load_sub_model
loaded_sub_model = load_method(os.path.join(cached_folder, name), **loading_kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\Python312\Lib\site-packages\huggingface_hub\utils_validators.py", line 114, in _inner_fn
return fn(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^
File "C:\Python312\Lib\site-packages\diffusers-0.30.0.dev0-py3.12.egg\diffusers\models\modeling_utils.py", line 784, in from_pretrained
accelerate.load_checkpoint_and_dispatch(
TypeError: load_checkpoint_and_dispatch() got an unexpected keyword argument 'strict'
Looks like you may be on a weird version of diffusers. Can you try out 0.29.2 by running pip install diffusers==0.29.2 ?
Installing collected packages: diffusers
Attempting uninstall: diffusers
Found existing installation: diffusers 0.29.2
Uninstalling diffusers-0.29.2:
Successfully uninstalled diffusers-0.29.2
Successfully installed diffusers-0.30.0.dev0
that solved the problem