Add new providers AnalogDiffusionProvider, NitroDiffusionProvider, and OpenJourneyProvider
This commit is contained in:
parent
7456becf9e
commit
14c3dbdcf1
|
@ -11,6 +11,9 @@ from .robertasquad2 import RobertaSquad2Provider
|
||||||
from .local import LocalProvider
|
from .local import LocalProvider
|
||||||
from .aihorde import AIHordeProvider
|
from .aihorde import AIHordeProvider
|
||||||
from .stablediffusion import StableDiffusionProvider
|
from .stablediffusion import StableDiffusionProvider
|
||||||
|
from .analogdiffusion import AnalogDiffusionProvider
|
||||||
|
from .nitrodiffusion import NitroDiffusionProvider
|
||||||
|
from .openjourney import OpenJourneyProvider
|
||||||
|
|
||||||
PROVIDERS = {
|
PROVIDERS = {
|
||||||
AIHordeProvider,
|
AIHordeProvider,
|
||||||
|
@ -23,6 +26,9 @@ PROVIDERS = {
|
||||||
GPT2Provider,
|
GPT2Provider,
|
||||||
LocalProvider,
|
LocalProvider,
|
||||||
StableDiffusionProvider,
|
StableDiffusionProvider,
|
||||||
|
AnalogDiffusionProvider,
|
||||||
|
NitroDiffusionProvider,
|
||||||
|
OpenJourneyProvider,
|
||||||
# StableBeluga2Provider,
|
# StableBeluga2Provider,
|
||||||
# HuggingFaceOpenAssistantSFT1PythiaProvider,
|
# HuggingFaceOpenAssistantSFT1PythiaProvider,
|
||||||
# RobertaSquad2Provider
|
# RobertaSquad2Provider
|
||||||
|
|
|
@ -0,0 +1,6 @@
|
||||||
|
from .basehfimage import BaseHFImageProvider
|
||||||
|
|
||||||
|
class AnalogDiffusionProvider(BaseHFImageProvider):
|
||||||
|
name = "Analog Diffusion"
|
||||||
|
provider = "wavymulder/Analog-Diffusion"
|
||||||
|
3
|
|
@ -2,6 +2,7 @@ providers_dir = join_paths(MODULE_DIR, 'providers')
|
||||||
|
|
||||||
providers_sources = [
|
providers_sources = [
|
||||||
'__init__.py',
|
'__init__.py',
|
||||||
|
'analogdiffusion.py',
|
||||||
'aihorde.py',
|
'aihorde.py',
|
||||||
'base.py',
|
'base.py',
|
||||||
'basehfimage.py',
|
'basehfimage.py',
|
||||||
|
@ -13,10 +14,12 @@ providers_sources = [
|
||||||
'gpt2.py',
|
'gpt2.py',
|
||||||
'hfbasechat.py',
|
'hfbasechat.py',
|
||||||
'local.py',
|
'local.py',
|
||||||
|
'nitrodiffusion.py',
|
||||||
'openai.py',
|
'openai.py',
|
||||||
'openaigpt35turbo.py',
|
'openaigpt35turbo.py',
|
||||||
'openaigpt4.py',
|
'openaigpt4.py',
|
||||||
'openassistantsft1pythia12b.py',
|
'openassistantsft1pythia12b.py',
|
||||||
|
'openjourney.py',
|
||||||
'petals.py',
|
'petals.py',
|
||||||
'provider_item.py',
|
'provider_item.py',
|
||||||
'stablebeluga2.py',
|
'stablebeluga2.py',
|
||||||
|
|
|
@ -0,0 +1,5 @@
|
||||||
|
from .basehfimage import BaseHFImageProvider
|
||||||
|
|
||||||
|
class NitroDiffusionProvider(BaseHFImageProvider):
|
||||||
|
name = "Nitro Diffusion"
|
||||||
|
provider = "nitrosocke/Nitro-Diffusion"
|
|
@ -0,0 +1,5 @@
|
||||||
|
from .basehfimage import BaseHFImageProvider
|
||||||
|
|
||||||
|
class OpenJourneyProvider(BaseHFImageProvider):
|
||||||
|
name = "Open Journey"
|
||||||
|
provider = "prompthero/openjourney-v4"
|
Loading…
Reference in New Issue