Spaces:
Build error
Build error
Commit ·
34643a5
1
Parent(s): 4236811
Update app.py
Browse files
app.py
CHANGED
|
@@ -88,6 +88,7 @@ def inference(prompt,canny_image,number,seed ):
|
|
| 88 |
#pipe.enable_model_cpu_offload()
|
| 89 |
# xformers
|
| 90 |
pipe.enable_xformers_memory_efficient_attention()
|
|
|
|
| 91 |
# Generator seed,
|
| 92 |
generator = torch.manual_seed(seed)
|
| 93 |
image_ = canny_image
|
|
|
|
| 88 |
#pipe.enable_model_cpu_offload()
|
| 89 |
# xformers
|
| 90 |
pipe.enable_xformers_memory_efficient_attention()
|
| 91 |
+
torch.cuda.is_available()
|
| 92 |
# Generator seed,
|
| 93 |
generator = torch.manual_seed(seed)
|
| 94 |
image_ = canny_image
|