Update modeling_transhla2.py
Browse files- modeling_transhla2.py +0 -1
modeling_transhla2.py
CHANGED
|
@@ -42,7 +42,6 @@ class LoraESM(nn.Module):
|
|
| 42 |
output = self.classifier(H)
|
| 43 |
return output, last_hidden_state
|
| 44 |
lora_esm = LoraESM()
|
| 45 |
-
lora_esm.load_state_dict(torch.load('Lora_ESM.pt'))
|
| 46 |
class TransHLA2(PreTrainedModel):
|
| 47 |
config_class = TransHLA2Config
|
| 48 |
|
|
|
|
| 42 |
output = self.classifier(H)
|
| 43 |
return output, last_hidden_state
|
| 44 |
lora_esm = LoraESM()
|
|
|
|
| 45 |
class TransHLA2(PreTrainedModel):
|
| 46 |
config_class = TransHLA2Config
|
| 47 |
|