semran1
/

test-ql

semran1 commited on Mar 9

Commit

ef5c938

verified ·

1 Parent(s): 9402361

Upload folder using huggingface_hub

Files changed (1) hide show

modeling_qllama.py CHANGED Viewed

@@ -1158,10 +1158,10 @@ class QLlamaForTokenClassification(QLlamaPreTrainedModel):
 __all__ = [
-    "LlamaForCausalLM",
-    "LlamaModel",
-    "LlamaPreTrainedModel",
-    "LlamaForSequenceClassification",
-    "LlamaForQuestionAnswering",
-    "LlamaForTokenClassification",
 ]

 __all__ = [
+    "QLlamaForCausalLM",
+    "QLlamaModel",
+    "QLlamaPreTrainedModel",
+    "QLlamaForSequenceClassification",
+    "QLlamaForQuestionAnswering",
+    "QLlamaForTokenClassification",
 ]