Hugging Face's logo

Spaces:
RED-AIGC
/
TDD
Running on Zero

App Files Community
BILLY12138 commited on
Commit
9080283
•
1 Parent(s): 6cfd3ee

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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":"RED-AIGC/TDD/sdxl_tdd_wo_adv_lora.safetensors",
17
- "TDD_adv":"RED-AIGC/TDD/sdxl_tdd_lora_weights.safetensors",
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():