forrest-vectara
commited on
Commit
•
8f6b0b5
1
Parent(s):
487d19f
remove unecessary `peft` dependency
Browse files- modeling_hhem_v2.py +2 -2
modeling_hhem_v2.py
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
import torch
|
2 |
|
3 |
-
from peft import PeftModel
|
4 |
from transformers import PreTrainedModel, AutoConfig, T5ForTokenClassification, AutoModel, AutoTokenizer, AutoModelForTokenClassification
|
5 |
|
6 |
from .configuration_hhem_v2 import HHEMv2Config
|
@@ -69,4 +69,4 @@ class HHEMv2ForSequenceClassification(PreTrainedModel):
|
|
69 |
raw_scores = transformed_probs[:, 1] # the probability of class 1
|
70 |
return raw_scores
|
71 |
|
72 |
-
|
|
|
1 |
import torch
|
2 |
|
3 |
+
#from peft import PeftModel
|
4 |
from transformers import PreTrainedModel, AutoConfig, T5ForTokenClassification, AutoModel, AutoTokenizer, AutoModelForTokenClassification
|
5 |
|
6 |
from .configuration_hhem_v2 import HHEMv2Config
|
|
|
69 |
raw_scores = transformed_probs[:, 1] # the probability of class 1
|
70 |
return raw_scores
|
71 |
|
72 |
+
|