Spaces:
Running
on
T4
Running
on
T4
Update app.py
Browse files
app.py
CHANGED
@@ -258,7 +258,7 @@ def genie (Model, Prompt, negative_prompt, height, width, scale, steps, seed, re
|
|
258 |
|
259 |
torch.cuda.empty_cache()
|
260 |
torch.cuda.max_memory_allocated(device=device)
|
261 |
-
sdxl = DiffusionPipeline.from_pretrained("stabilityai/
|
262 |
sdxl.enable_xformers_memory_efficient_attention()
|
263 |
sdxl = sdxl.to(device)
|
264 |
torch.cuda.empty_cache()
|
|
|
258 |
|
259 |
torch.cuda.empty_cache()
|
260 |
torch.cuda.max_memory_allocated(device=device)
|
261 |
+
sdxl = DiffusionPipeline.from_pretrained("stabilityai/cosxl", torch_dtype=torch.float16, variant="fp16", use_safetensors=True)
|
262 |
sdxl.enable_xformers_memory_efficient_attention()
|
263 |
sdxl = sdxl.to(device)
|
264 |
torch.cuda.empty_cache()
|