Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -27,7 +27,7 @@ torch.cuda.empty_cache()
|
|
27 |
def genie (prompt, negative_prompt, scale, steps, seed):
|
28 |
torch.cuda.empty_cache()
|
29 |
generator = torch.Generator(device=device).manual_seed(seed)
|
30 |
-
int_image = pipe(prompt, negative_prompt=negative_prompt, num_inference_steps=steps, guidance_scale=scale, num_images_per_prompt=1, generator=generator, width=
|
31 |
torch.cuda.empty_cache()
|
32 |
image = refiner(prompt=prompt, image=int_image).images[0]
|
33 |
return image
|
|
|
27 |
def genie (prompt, negative_prompt, scale, steps, seed):
|
28 |
torch.cuda.empty_cache()
|
29 |
generator = torch.Generator(device=device).manual_seed(seed)
|
30 |
+
int_image = pipe(prompt, negative_prompt=negative_prompt, num_inference_steps=steps, guidance_scale=scale, num_images_per_prompt=1, generator=generator, width=768, height=768, output_type="latent").images
|
31 |
torch.cuda.empty_cache()
|
32 |
image = refiner(prompt=prompt, image=int_image).images[0]
|
33 |
return image
|