Customized diffusers with model search and other functions.
Project description
auto_diffusers
CONTENTS
About The Project
Enhance the functionality of diffusers.
- Search models from huggingface and Civitai. (etc..)
How to use
pip install diffusers
pip install auto_diffusers
from diffusers import StableDiffusionPipeline
from auto_diffusers import run_search
model_path = run_search(<keyword>, auto=True, download=False)
pipe = StableDIffusionPipeline.from_single_file(model_path)
Description
Arguments of
run_search
Name | Type | Default | Input Available | Description |
---|---|---|---|---|
search_word | string | ー | Details | Keywords to search models |
auto | bool | True | ー | Minimize user input by selecting the highest-rated models. |
download | bool | False | ー | Returns the path where the file was downloaded. |
model_type | string | "Checkpoint" | 1. Checkpoint 2. TextualInversion 3. Hypernetwork 4. AestheticGradient 5. LORA 6. Controlnet 7. Poses |
Valid only in Civitai. |
return_path | bool | True | ー | Returns only the path or [model_path, status_dict] . |
branch | string | "main" | ー | Specify the branches of huggingface and civitai. |
local_file_only | bool | False | ー | Search local folders only. |
search_word
Type | Description |
---|---|
keyword | Keywords to search model |
url | Can be any URL other than huggingface or Civitai. |
Local directory or file path | Search for files with the extensions: .safetensors , .ckpt , .bin |
huggingface path | The following format: < creator > / < repo > |
License
In accordance with BSD-3-Clause license
Acknowledgement
I have used open source resources and free tools in the creation of this project.
I would like to take this opportunity to thank the open source community and those who provided free tools.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
auto_diffusers-1.2.2.dev5.tar.gz
(23.6 kB
view hashes)
Built Distribution
Close
Hashes for auto_diffusers-1.2.2.dev5.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | 3f7d131984664c4878b25de7fe07b43270af276bdb59b60dd8f4490e3d97debf |
|
MD5 | 10253fa87fb1d9fcbd802a8f5857e144 |
|
BLAKE2b-256 | 90152c1c349f3200b6af7ec8fab2c102e4922e5fcbe133d57002d6d95803c18f |
Close
Hashes for auto_diffusers-1.2.2.dev5-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | d5731d506a29d87d2fa834758705d450358ddb73377aec2bfe9b20929bb0505b |
|
MD5 | 03e0e8b51b0281f50141af67056f448a |
|
BLAKE2b-256 | c9a7e7fdff1bd9d3bce34f86d40eec6cb47574282a9ebc0a41b2745b2eed95c8 |