Update app.py
Browse files
app.py
CHANGED
|
@@ -13,8 +13,8 @@ device = "cuda"
|
|
| 13 |
#device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 14 |
|
| 15 |
ACC_lora={
|
| 16 |
-
"TDD":"
|
| 17 |
-
"TDD_adv":"
|
| 18 |
}
|
| 19 |
|
| 20 |
if torch.cuda.is_available():
|
|
|
|
| 13 |
#device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 14 |
|
| 15 |
ACC_lora={
|
| 16 |
+
"TDD":"sdxl_tdd_wo_adv_lora.safetensors",
|
| 17 |
+
"TDD_adv":"sdxl_tdd_lora_weights.safetensors",
|
| 18 |
}
|
| 19 |
|
| 20 |
if torch.cuda.is_available():
|