Update README.md
Browse files
README.md
CHANGED
|
@@ -54,7 +54,6 @@ class OptILMClassifier(nn.Module):
|
|
| 54 |
logits = self.classifier(combined_input)
|
| 55 |
return logits
|
| 56 |
|
| 57 |
-
|
| 58 |
def load_optillm_model():
|
| 59 |
device = torch.device("mps" if torch.backends.mps.is_available() else "cuda" if torch.cuda.is_available() else "cpu")
|
| 60 |
# Load the base model
|
|
|
|
| 54 |
logits = self.classifier(combined_input)
|
| 55 |
return logits
|
| 56 |
|
|
|
|
| 57 |
def load_optillm_model():
|
| 58 |
device = torch.device("mps" if torch.backends.mps.is_available() else "cuda" if torch.cuda.is_available() else "cpu")
|
| 59 |
# Load the base model
|