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 .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
|
||||
|
|
|
@ -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 = [
|
||||
'__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',
|
||||
|
|
|
@ -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