Add new providers AnalogDiffusionProvider, NitroDiffusionProvider, and OpenJourneyProvider

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

src/providers/analogdiffusion.py

@@ -0,0 +1,6 @@
+from .basehfimage import BaseHFImageProvider
+
+class AnalogDiffusionProvider(BaseHFImageProvider):
+    name = "Analog Diffusion"
+    provider = "wavymulder/Analog-Diffusion"
+3

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',

src/providers/nitrodiffusion.py

@@ -0,0 +1,5 @@
+from .basehfimage import BaseHFImageProvider
+
+class NitroDiffusionProvider(BaseHFImageProvider):
+    name = "Nitro Diffusion"
+    provider = "nitrosocke/Nitro-Diffusion"

src/providers/openjourney.py

@@ -0,0 +1,5 @@
+from .basehfimage import BaseHFImageProvider
+
+class OpenJourneyProvider(BaseHFImageProvider):
+    name = "Open Journey"
+    provider = "prompthero/openjourney-v4"