Spaces:
Sleeping
Sleeping
wip
Browse files
app.py
CHANGED
@@ -12,7 +12,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
12 |
torch.cuda.max_memory_allocated(device=device)
|
13 |
# pipe = DiffusionPipeline.from_pretrained("stabilityai/sdxl-turbo", torch_dtype=torch.float16, variant="fp16", use_safetensors=True)
|
14 |
pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16)
|
15 |
-
pipe.enable_xformers_memory_efficient_attention()
|
16 |
pipe = pipe.to(device)
|
17 |
# else:
|
18 |
# pipe = DiffusionPipeline.from_pretrained("stabilityai/sdxl-turbo", use_safetensors=True)
|
|
|
12 |
torch.cuda.max_memory_allocated(device=device)
|
13 |
# pipe = DiffusionPipeline.from_pretrained("stabilityai/sdxl-turbo", torch_dtype=torch.float16, variant="fp16", use_safetensors=True)
|
14 |
pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16)
|
15 |
+
# pipe.enable_xformers_memory_efficient_attention()
|
16 |
pipe = pipe.to(device)
|
17 |
# else:
|
18 |
# pipe = DiffusionPipeline.from_pretrained("stabilityai/sdxl-turbo", use_safetensors=True)
|