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 |
|