yuxinjiang11 commited on
Commit
e97455b
·
verified ·
1 Parent(s): 41e483a

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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/ip_adapter_0.bin", "ip_ckpt_path"),
125
- ("checkpoint/att.bin", "att_ckpt_path")
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: