Spaces:
Runtime error
Runtime error
disable xformers again
Browse files
app.py
CHANGED
@@ -2,6 +2,7 @@ import gradio as gr
|
|
2 |
from diffusers import StableDiffusionControlNetPipeline, ControlNetModel
|
3 |
from diffusers import UniPCMultistepScheduler
|
4 |
import torch
|
|
|
5 |
import torchvision
|
6 |
import torchvision.transforms as T
|
7 |
#from torchvision.transforms import v2 as T2
|
@@ -31,7 +32,7 @@ pipe = StableDiffusionControlNetPipeline.from_pretrained(
|
|
31 |
).to("cuda")
|
32 |
pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
|
33 |
#pipe.enable_model_cpu_offload()
|
34 |
-
pipe.enable_xformers_memory_efficient_attention()
|
35 |
|
36 |
generator = torch.manual_seed(0)
|
37 |
|
|
|
2 |
from diffusers import StableDiffusionControlNetPipeline, ControlNetModel
|
3 |
from diffusers import UniPCMultistepScheduler
|
4 |
import torch
|
5 |
+
torch.backends.cuda.matmul.allow_tf32 = True
|
6 |
import torchvision
|
7 |
import torchvision.transforms as T
|
8 |
#from torchvision.transforms import v2 as T2
|
|
|
32 |
).to("cuda")
|
33 |
pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
|
34 |
#pipe.enable_model_cpu_offload()
|
35 |
+
#pipe.enable_xformers_memory_efficient_attention()
|
36 |
|
37 |
generator = torch.manual_seed(0)
|
38 |
|