Spaces:
Sleeping
Sleeping
Commit
·
f67fd7b
1
Parent(s):
6269d98
reduced the number of concepts and steps for inference
Browse files
app.py
CHANGED
@@ -12,11 +12,11 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
12 |
|
13 |
# List of concept embeddings compatible with SD v1.4
|
14 |
concepts = [
|
15 |
-
"sd-concepts-library/cat-toy"
|
16 |
-
"sd-concepts-library/disco-diffusion-style",
|
17 |
-
"sd-concepts-library/modern-disney-style",
|
18 |
-
"sd-concepts-library/charliebo-artstyle",
|
19 |
-
"sd-concepts-library/redshift-render-style"
|
20 |
]
|
21 |
|
22 |
def download_concept_embedding(concept_name):
|
@@ -83,7 +83,7 @@ def generate_images(prompt):
|
|
83 |
with autocast(device):
|
84 |
image = pipe(
|
85 |
concept_prompt,
|
86 |
-
num_inference_steps=
|
87 |
generator=generator,
|
88 |
guidance_scale=7.5
|
89 |
).images[0]
|
|
|
12 |
|
13 |
# List of concept embeddings compatible with SD v1.4
|
14 |
concepts = [
|
15 |
+
"sd-concepts-library/cat-toy"
|
16 |
+
# "sd-concepts-library/disco-diffusion-style",
|
17 |
+
# "sd-concepts-library/modern-disney-style",
|
18 |
+
# "sd-concepts-library/charliebo-artstyle",
|
19 |
+
# "sd-concepts-library/redshift-render-style"
|
20 |
]
|
21 |
|
22 |
def download_concept_embedding(concept_name):
|
|
|
83 |
with autocast(device):
|
84 |
image = pipe(
|
85 |
concept_prompt,
|
86 |
+
num_inference_steps=20,
|
87 |
generator=generator,
|
88 |
guidance_scale=7.5
|
89 |
).images[0]
|