Update app.py
Browse files
app.py
CHANGED
@@ -42,7 +42,7 @@ def genie (prompt, negative_prompt, height, width, scale, steps, seed, upscaler)
|
|
42 |
pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-refiner-0.9", torch_dtype=torch.float16, variant="fp16", use_safetensors=True)
|
43 |
pipe = pipe.to(device)
|
44 |
pipe.enable_xformers_memory_efficient_attention()
|
45 |
-
image = pipe(prompt=prompt, image=int_image).images[0]
|
46 |
torch.cuda.empty_cache()
|
47 |
return (image, image)
|
48 |
|
|
|
42 |
pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-refiner-0.9", torch_dtype=torch.float16, variant="fp16", use_safetensors=True)
|
43 |
pipe = pipe.to(device)
|
44 |
pipe.enable_xformers_memory_efficient_attention()
|
45 |
+
image = pipe(prompt=prompt, negative_prompt=negative_prompt, image=int_image).images[0]
|
46 |
torch.cuda.empty_cache()
|
47 |
return (image, image)
|
48 |
|