bergr7f commited on
Commit
eb8daf9
·
1 Parent(s): 2875a07

Disable flash attention

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from flow_judge import Hf, FlowJudge, EvalInput
5
  from flow_judge.metrics import CustomMetric, RubricItem
6
 
7
  try:
8
- model = Hf()
9
  except Exception as e:
10
  raise RuntimeError(f"Failed to initialize Hf Model: {e}")
11
 
 
5
  from flow_judge.metrics import CustomMetric, RubricItem
6
 
7
  try:
8
+ model = Hf(flash_attn=False)
9
  except Exception as e:
10
  raise RuntimeError(f"Failed to initialize Hf Model: {e}")
11