ykae commited on
Commit
f18061b
·
verified ·
1 Parent(s): 2943dd7

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -94,8 +94,8 @@ model = AutoModelForSequenceClassification.from_pretrained(
94
  ).to(device)
95
 
96
  # 2. Performance Optimization (Mandatory for Monarch Speed)
97
- torch.set_float32_matmul_precision('high')
98
- model = torch.compile(model, mode="max-autotune")
99
  model.eval()
100
 
101
  # 3. Load MNLI Validation Set
 
94
  ).to(device)
95
 
96
  # 2. Performance Optimization (Mandatory for Monarch Speed)
97
+ # torch.set_float32_matmul_precision('high')
98
+ # model = torch.compile(model, mode="max-autotune")
99
  model.eval()
100
 
101
  # 3. Load MNLI Validation Set