Spaces:
Running
Running
Upload app.py
Browse files
app.py
CHANGED
|
@@ -121,8 +121,8 @@ class SingleAnomalyGenerator:
|
|
| 121 |
# Load weights (download from cloud repo to memory, avoid any disk usage)
|
| 122 |
print("Loading weights into memory...")
|
| 123 |
weight_files = [
|
| 124 |
-
("checkpoint/
|
| 125 |
-
("checkpoint/
|
| 126 |
]
|
| 127 |
for filename, attr_name in weight_files:
|
| 128 |
try:
|
|
|
|
| 121 |
# Load weights (download from cloud repo to memory, avoid any disk usage)
|
| 122 |
print("Loading weights into memory...")
|
| 123 |
weight_files = [
|
| 124 |
+
("checkpoint/anomagic.bin", "ip_ckpt_path"),
|
| 125 |
+
("checkpoint/attention_module.bin", "att_ckpt_path")
|
| 126 |
]
|
| 127 |
for filename, attr_name in weight_files:
|
| 128 |
try:
|