Update app.py
Browse files
app.py
CHANGED
|
@@ -46,7 +46,7 @@ vae_dir = Path(model_path) / "vae"
|
|
| 46 |
unet_dir = Path(model_path) / "unet"
|
| 47 |
scheduler_dir = Path(model_path) / "scheduler"
|
| 48 |
|
| 49 |
-
device = torch.device("cuda")
|
| 50 |
|
| 51 |
request_log = []
|
| 52 |
|
|
|
|
| 46 |
unet_dir = Path(model_path) / "unet"
|
| 47 |
scheduler_dir = Path(model_path) / "scheduler"
|
| 48 |
|
| 49 |
+
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 50 |
|
| 51 |
request_log = []
|
| 52 |
|