Text Classification
Transformers
Safetensors
English
HHEMv2Config
custom_code
forrest-vectara commited on
Commit
8f6b0b5
1 Parent(s): 487d19f

remove unecessary `peft` dependency

Browse files
Files changed (1) hide show
  1. 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
+