Update app.py
Browse files
app.py
CHANGED
@@ -39,7 +39,7 @@ def genie (prompt, negative_prompt, height, width, scale, steps, seed, upscaler)
|
|
39 |
else:
|
40 |
torch.cuda.empty_cache()
|
41 |
torch.cuda.max_memory_allocated(device=device)
|
42 |
-
pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-refiner-0
|
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]
|
|
|
39 |
else:
|
40 |
torch.cuda.empty_cache()
|
41 |
torch.cuda.max_memory_allocated(device=device)
|
42 |
+
pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-refiner-1.0", 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]
|