Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -58,7 +58,7 @@ def genie (Model, Prompt, negative_prompt, height, width, scale, steps, seed, re
|
|
58 |
|
59 |
torch.cuda.empty_cache()
|
60 |
torch.cuda.max_memory_allocated(device=device)
|
61 |
-
sdxl =
|
62 |
sdxl.enable_xformers_memory_efficient_attention()
|
63 |
sdxl = sdxl.to(device)
|
64 |
torch.cuda.empty_cache()
|
|
|
58 |
|
59 |
torch.cuda.empty_cache()
|
60 |
torch.cuda.max_memory_allocated(device=device)
|
61 |
+
sdxl = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3-medium-diffusers", torch_dtype=torch.float16, )#variant="fp16", use_safetensors=True)
|
62 |
sdxl.enable_xformers_memory_efficient_attention()
|
63 |
sdxl = sdxl.to(device)
|
64 |
torch.cuda.empty_cache()
|