Spaces:
Paused
Paused
HanningChen
commited on
Commit
·
1a6c72f
1
Parent(s):
11d6c56
Fix cross attention
Browse files- webui/runner.py +2 -1
webui/runner.py
CHANGED
|
@@ -421,7 +421,8 @@ class ModelRunner:
|
|
| 421 |
# - default => cross_attention True
|
| 422 |
# - HDC => cross_attention False
|
| 423 |
# If your actual training differs, change this rule OR pass it from app.py.
|
| 424 |
-
cross_attention = (score_function != "HDC")
|
|
|
|
| 425 |
|
| 426 |
with self._lock:
|
| 427 |
img = Image.open(image_path).convert("RGB")
|
|
|
|
| 421 |
# - default => cross_attention True
|
| 422 |
# - HDC => cross_attention False
|
| 423 |
# If your actual training differs, change this rule OR pass it from app.py.
|
| 424 |
+
# cross_attention = (score_function != "HDC")
|
| 425 |
+
cross_attention = True
|
| 426 |
|
| 427 |
with self._lock:
|
| 428 |
img = Image.open(image_path).convert("RGB")
|