Spaces:
Sleeping
Sleeping
Commit
•
813b8fa
1
Parent(s):
9986ae7
bug fixes
Browse files
app.py
CHANGED
@@ -85,7 +85,7 @@ def get_output_embeds(input_embeddings):
|
|
85 |
def generate_with_embs(text_embeddings, seed, max_length):
|
86 |
height = 512 # default height of Stable Diffusion
|
87 |
width = 512 # default width of Stable Diffusion
|
88 |
-
num_inference_steps =
|
89 |
guidance_scale = 7.5 # Scale for classifier-free guidance
|
90 |
generator = torch.manual_seed(seed) # Seed generator to create the inital latent noise
|
91 |
batch_size = 1
|
@@ -171,7 +171,7 @@ def generate_image_on_loss(prompt, seed):
|
|
171 |
|
172 |
height = 64 # default height of Stable Diffusion
|
173 |
width = 64 # default width of Stable Diffusion
|
174 |
-
num_inference_steps =
|
175 |
guidance_scale = 8 # Scale for classifier-free guidance
|
176 |
generator = torch.manual_seed(64) # Seed generator to create the inital latent noise
|
177 |
batch_size = 1
|
@@ -284,4 +284,4 @@ demo = gr.Interface(
|
|
284 |
]
|
285 |
)
|
286 |
|
287 |
-
demo.launch()
|
|
|
85 |
def generate_with_embs(text_embeddings, seed, max_length):
|
86 |
height = 512 # default height of Stable Diffusion
|
87 |
width = 512 # default width of Stable Diffusion
|
88 |
+
num_inference_steps = 30 # Number of denoising steps
|
89 |
guidance_scale = 7.5 # Scale for classifier-free guidance
|
90 |
generator = torch.manual_seed(seed) # Seed generator to create the inital latent noise
|
91 |
batch_size = 1
|
|
|
171 |
|
172 |
height = 64 # default height of Stable Diffusion
|
173 |
width = 64 # default width of Stable Diffusion
|
174 |
+
num_inference_steps = 50 # Number of denoising steps
|
175 |
guidance_scale = 8 # Scale for classifier-free guidance
|
176 |
generator = torch.manual_seed(64) # Seed generator to create the inital latent noise
|
177 |
batch_size = 1
|
|
|
284 |
]
|
285 |
)
|
286 |
|
287 |
+
demo.launch(debug=True)
|