Skip to main content

Customized diffusers with model search and other functions.

Project description

auto_diffusers

GitHub license

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


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.1.0.tar.gz (23.4 kB view hashes)

Uploaded Source

Built Distribution

auto_diffusers-1.1.0-py3-none-any.whl (30.1 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page