Commit
•
8ea1b21
1
Parent(s):
23ad37a
Update app.py
Browse files
app.py
CHANGED
|
@@ -13,6 +13,7 @@ import random
|
|
| 13 |
import time
|
| 14 |
|
| 15 |
dtype = torch.bfloat16
|
|
|
|
| 16 |
|
| 17 |
# Load LoRAs from JSON file
|
| 18 |
with open('loras.json', 'r') as f:
|
|
|
|
| 13 |
import time
|
| 14 |
|
| 15 |
dtype = torch.bfloat16
|
| 16 |
+
device = "cuda"
|
| 17 |
|
| 18 |
# Load LoRAs from JSON file
|
| 19 |
with open('loras.json', 'r') as f:
|