chunzhi/stable-diffusion-cli-on-modal
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #23 from hodanov/feature/refactoring

Browse Source 
Modify a little.
This commit is contained in:
hodanov 2023-07-18 23:15:56 +09:00 committed by GitHub
commit ffbcb8aae0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
setup_files/__main__.py
View File

@ -1,11 +1,7 @@
from __future__ import annotations from __future__ import annotations
from setup import stub from setup import stub
from txt2img import StableDiffusion from txt2img import StableDiffusion, new_stable_diffusion
def new_stable_diffusion() -> StableDiffusion:
return StableDiffusion()
@stub.function(gpu="A10G") @stub.function(gpu="A10G")

4
setup_files/txt2img.py
View File

@ -14,6 +14,10 @@ from setup import (BASE_CACHE_PATH, BASE_CACHE_PATH_CONTROLNET,
stub) stub)
def new_stable_diffusion() -> StableDiffusionInterface:
return StableDiffusion()
class StableDiffusionInterface(metaclass=abc.ABCMeta): class StableDiffusionInterface(metaclass=abc.ABCMeta):
""" """
A StableDiffusionInterface is an interface that will be used for StableDiffusion class creation. A StableDiffusionInterface is an interface that will be used for StableDiffusion class creation.