DisgustingOzil
commited on
Commit
•
bcbfbad
1
Parent(s):
844d72c
updated handler.py
Browse files- handler.py +1 -1
handler.py
CHANGED
@@ -3,7 +3,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
|
|
3 |
|
4 |
class EndpointHandler:
|
5 |
def __init__(self, path=""):
|
6 |
-
model_id = "DisgustingOzil/Academic-
|
7 |
load_in_4bit = True
|
8 |
self.tokenizer = AutoTokenizer.from_pretrained(model_id)
|
9 |
self.model = AutoModelForCausalLM.from_pretrained(model_id, load_in_4bit=load_in_4bit)
|
|
|
3 |
|
4 |
class EndpointHandler:
|
5 |
def __init__(self, path=""):
|
6 |
+
model_id = "DisgustingOzil/Academic-ShortQA-Generator"
|
7 |
load_in_4bit = True
|
8 |
self.tokenizer = AutoTokenizer.from_pretrained(model_id)
|
9 |
self.model = AutoModelForCausalLM.from_pretrained(model_id, load_in_4bit=load_in_4bit)
|