Spaces:
Sleeping
Sleeping
Yaron Koresh
commited on
Update app.py
Browse files
app.py
CHANGED
|
@@ -12,9 +12,9 @@ from diffusers import AutoPipelineForText2Image
|
|
| 12 |
#from tempfile import NamedTemporaryFile
|
| 13 |
from pathos.threading import ThreadPool as Pool
|
| 14 |
|
| 15 |
-
|
| 16 |
#model_id = "kandinsky-community/kandinsky-3"
|
| 17 |
-
model_id = "stabilityai/stable-diffusion-3-medium-diffusers"
|
| 18 |
|
| 19 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 20 |
if torch.cuda.is_available():
|
|
@@ -91,7 +91,7 @@ def Piper(_do,_dont):
|
|
| 91 |
width=384,
|
| 92 |
negative_prompt=_dont,
|
| 93 |
num_inference_steps=100,
|
| 94 |
-
guidance_scale=3
|
| 95 |
)
|
| 96 |
|
| 97 |
def infer(prompt_en,prompt2_en):
|
|
|
|
| 12 |
#from tempfile import NamedTemporaryFile
|
| 13 |
from pathos.threading import ThreadPool as Pool
|
| 14 |
|
| 15 |
+
model_id = "runwayml/stable-diffusion-v1-5"
|
| 16 |
#model_id = "kandinsky-community/kandinsky-3"
|
| 17 |
+
#model_id = "stabilityai/stable-diffusion-3-medium-diffusers"
|
| 18 |
|
| 19 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 20 |
if torch.cuda.is_available():
|
|
|
|
| 91 |
width=384,
|
| 92 |
negative_prompt=_dont,
|
| 93 |
num_inference_steps=100,
|
| 94 |
+
guidance_scale=3
|
| 95 |
)
|
| 96 |
|
| 97 |
def infer(prompt_en,prompt2_en):
|