change
Browse files
app.py
CHANGED
@@ -26,7 +26,6 @@ import gc
|
|
26 |
from datetime import datetime
|
27 |
from huggingface_hub import snapshot_download
|
28 |
log = logging.getLogger()
|
29 |
-
|
30 |
device = "cpu"
|
31 |
if torch.cuda.is_available():
|
32 |
device = "cuda"
|
|
|
26 |
from datetime import datetime
|
27 |
from huggingface_hub import snapshot_download
|
28 |
log = logging.getLogger()
|
|
|
29 |
device = "cpu"
|
30 |
if torch.cuda.is_available():
|
31 |
device = "cuda"
|