diff --git a/src/providers/__init__.py b/src/providers/__init__.py index 2a9df78..c5a1064 100644 --- a/src/providers/__init__.py +++ b/src/providers/__init__.py @@ -11,6 +11,9 @@ from .robertasquad2 import RobertaSquad2Provider from .local import LocalProvider from .aihorde import AIHordeProvider from .stablediffusion import StableDiffusionProvider +from .analogdiffusion import AnalogDiffusionProvider +from .nitrodiffusion import NitroDiffusionProvider +from .openjourney import OpenJourneyProvider PROVIDERS = { AIHordeProvider, @@ -23,6 +26,9 @@ PROVIDERS = { GPT2Provider, LocalProvider, StableDiffusionProvider, + AnalogDiffusionProvider, + NitroDiffusionProvider, + OpenJourneyProvider, # StableBeluga2Provider, # HuggingFaceOpenAssistantSFT1PythiaProvider, # RobertaSquad2Provider diff --git a/src/providers/analogdiffusion.py b/src/providers/analogdiffusion.py new file mode 100644 index 0000000..fc6fa75 --- /dev/null +++ b/src/providers/analogdiffusion.py @@ -0,0 +1,6 @@ +from .basehfimage import BaseHFImageProvider + +class AnalogDiffusionProvider(BaseHFImageProvider): + name = "Analog Diffusion" + provider = "wavymulder/Analog-Diffusion" +3 diff --git a/src/providers/meson.build b/src/providers/meson.build index 4de5d35..4009db5 100644 --- a/src/providers/meson.build +++ b/src/providers/meson.build @@ -2,6 +2,7 @@ providers_dir = join_paths(MODULE_DIR, 'providers') providers_sources = [ '__init__.py', + 'analogdiffusion.py', 'aihorde.py', 'base.py', 'basehfimage.py', @@ -13,10 +14,12 @@ providers_sources = [ 'gpt2.py', 'hfbasechat.py', 'local.py', + 'nitrodiffusion.py', 'openai.py', 'openaigpt35turbo.py', 'openaigpt4.py', 'openassistantsft1pythia12b.py', + 'openjourney.py', 'petals.py', 'provider_item.py', 'stablebeluga2.py', diff --git a/src/providers/nitrodiffusion.py b/src/providers/nitrodiffusion.py new file mode 100644 index 0000000..42786e4 --- /dev/null +++ b/src/providers/nitrodiffusion.py @@ -0,0 +1,5 @@ +from .basehfimage import BaseHFImageProvider + +class NitroDiffusionProvider(BaseHFImageProvider): + name = "Nitro Diffusion" + provider = "nitrosocke/Nitro-Diffusion" \ No newline at end of file diff --git a/src/providers/openjourney.py b/src/providers/openjourney.py new file mode 100644 index 0000000..847c577 --- /dev/null +++ b/src/providers/openjourney.py @@ -0,0 +1,5 @@ +from .basehfimage import BaseHFImageProvider + +class OpenJourneyProvider(BaseHFImageProvider): + name = "Open Journey" + provider = "prompthero/openjourney-v4" \ No newline at end of file