wath5
/

kgl_lmsys_pref_classif

Text Classification

text-generation-inference

Inference Endpoints

8-bit precision

Model card Files Files and versions Community

wath5 commited on 8 days ago

Commit

cd24909

•

1 Parent(s): 421c26f

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -45,7 +45,7 @@ Mimic human preference given a query and 2 different answers.
 import torch
 from torch import nn
 from torch.nn import CrossEntropyLoss, MSELoss, BCEWithLogitsLoss
-from transformers import Gemma2PreTrainedModel,Gemma2Model
 from transformers.modeling_outputs import SequenceClassifierOutputWithPast
 from typing import Optional, List, Union, Tuple
 from dataclasses import dataclass

 import torch
 from torch import nn
 from torch.nn import CrossEntropyLoss, MSELoss, BCEWithLogitsLoss
+from transformers import Gemma2PreTrainedModel,Gemma2Model, Cache, AutoTokenizer
 from transformers.modeling_outputs import SequenceClassifierOutputWithPast
 from typing import Optional, List, Union, Tuple
 from dataclasses import dataclass