HanningChen commited on
Commit
1a6c72f
·
1 Parent(s): 11d6c56

Fix cross attention

Browse files
Files changed (1) hide show
  1. 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")