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 model_search
path = model_search(
<keyword>,
auto = True,
model_format="diffusers",
download = False
)
pipe = StableDiffusionPipeline.from_pretrained(path)
# or
path = model_search(
<keyword>,
auto = True,
model_format="single_file",
download = False
)
pipe = StableDIffusionPipeline.from_single_file(path)
Example
pip install --quiet diffusers
pip install --quiet auto_diffusers
from diffusers import StableDiffusionPipeline
from IPython.display import display
from auto_diffusers import model_search
model_path = model_search(
"Any",
auto=True,
model_format="diffusers",
download=False
)
pipe = StableDiffusionPipeline.from_pretrained(model_path).to("cuda")
image = pipe("Mt. Fuji").images[0]
print(f"model_path: {model_path}")
display(image)
Description
Arguments of
model_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_format | string | "single_file" | all ,diffusers ,single_file |
Specifies the format of the model. Details |
model_type | string | "Checkpoint" | Checkpoint ,TextualInversion ,Hypernetwork ,AestheticGradient ,LORA ,Controlnet ,Poses |
Valid only in Civitai. |
include_params | bool | False | ー | Returns the model path or a dictionary with parameters. |
branch | string | "main" | ー | Specify the branches of huggingface and civitai. |
local_file_only | bool | False | ー | Search local folders only. In the case of auto , files with names similar to search_word will be given priority. |
hf_token | string | None | ー | Token used for authentication with Hugging Face. |
civitai_token | string | None | ー | Token used for authentication with Civitai. |
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 > |
model_format
Argument | Description |
---|---|
all | In auto, multifolder diffusers format checkpoint takes precedence |
single_file | Only single file checkpoint are searched. |
diffusers | Search only for multifolder diffusers format checkpoint Note that only the huggingface is searched for, since it is not in civitai. |
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.7.7.tar.gz
(23.3 kB
view hashes)
Built Distribution
Close
Hashes for auto_diffusers-1.7.7-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 7ea38fd312e8c4f749eedc13f0e923c59faee908e53e233292dcc5ee7be36107 |
|
MD5 | ee2718f3ddded83c6cc83fe2d8ea5b1c |
|
BLAKE2b-256 | eb4c29a421eb2b1f97c3f06150da068e7f635bcbc4206c912294d4f6c1b92e03 |