Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -60,7 +60,7 @@ USE_TORCH_COMPILE = os.getenv("USE_TORCH_COMPILE", "0") == "1"
|
|
60 |
ENABLE_CPU_OFFLOAD = 0
|
61 |
BATCH_SIZE = int(os.getenv("BATCH_SIZE", "1"))
|
62 |
|
63 |
-
device = torch.device("cuda:0"
|
64 |
|
65 |
style_list = [
|
66 |
{
|
|
|
60 |
ENABLE_CPU_OFFLOAD = 0
|
61 |
BATCH_SIZE = int(os.getenv("BATCH_SIZE", "1"))
|
62 |
|
63 |
+
device = torch.device("cuda:0")
|
64 |
|
65 |
style_list = [
|
66 |
{
|