Spaces:
Runtime error
Runtime error
Commit
·
b12de1e
1
Parent(s):
3c2cdca
update
Browse files
app.py
CHANGED
|
@@ -13,6 +13,7 @@ from PIL import Image
|
|
| 13 |
from tqdm.auto import tqdm
|
| 14 |
from datetime import datetime
|
| 15 |
from utils.gradio_utils import is_torch2_available
|
|
|
|
| 16 |
if is_torch2_available():
|
| 17 |
from utils.gradio_utils import \
|
| 18 |
AttnProcessor2_0 as AttnProcessor
|
|
|
|
| 13 |
from tqdm.auto import tqdm
|
| 14 |
from datetime import datetime
|
| 15 |
from utils.gradio_utils import is_torch2_available
|
| 16 |
+
import devicetorch
|
| 17 |
if is_torch2_available():
|
| 18 |
from utils.gradio_utils import \
|
| 19 |
AttnProcessor2_0 as AttnProcessor
|