Spaces:
Sleeping
Sleeping
Gainward777
commited on
Update sd/zerogpu_controller.py
Browse files- sd/zerogpu_controller.py +1 -6
sd/zerogpu_controller.py
CHANGED
@@ -1,4 +1,3 @@
|
|
1 |
-
#from ui.gradio_ui import ui
|
2 |
import spaces
|
3 |
|
4 |
from sd.prompt_helper import helper
|
@@ -6,9 +5,7 @@ from PIL import Image
|
|
6 |
|
7 |
from sd.utils.utils import *
|
8 |
from utils.utils import sketch_process, prompt_preprocess
|
9 |
-
#from sd.sd_controller import Controller
|
10 |
|
11 |
-
#controller=Controller()
|
12 |
|
13 |
MODELS_NAMES=["cagliostrolab/animagine-xl-3.1",
|
14 |
"stabilityai/stable-diffusion-xl-base-1.0"]
|
@@ -83,6 +80,4 @@ def get_second_result(img, prompt, negative_prompt,
|
|
83 |
|
84 |
|
85 |
def get_help_w_prompt(img):
|
86 |
-
return helper.get_help(img["composite"])
|
87 |
-
|
88 |
-
#ui(get_first_result, get_second_result) #controller)
|
|
|
|
|
1 |
import spaces
|
2 |
|
3 |
from sd.prompt_helper import helper
|
|
|
5 |
|
6 |
from sd.utils.utils import *
|
7 |
from utils.utils import sketch_process, prompt_preprocess
|
|
|
8 |
|
|
|
9 |
|
10 |
MODELS_NAMES=["cagliostrolab/animagine-xl-3.1",
|
11 |
"stabilityai/stable-diffusion-xl-base-1.0"]
|
|
|
80 |
|
81 |
|
82 |
def get_help_w_prompt(img):
|
83 |
+
return helper.get_help(img["composite"])
|
|
|
|