Itamarl commited on
Commit
1dc374d
·
1 Parent(s): 45045e3

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -0
handler.py CHANGED
@@ -7,6 +7,7 @@ from datetime import datetime
7
  class EndpointHandler():
8
 
9
  def __init__(self, path=""):
 
10
  self.model = transformers.AutoModelForCausalLM.from_pretrained(path,
11
  #"/Users/itamarlevi/Downloads/my_repo_hf/hf/mpt-7b/venv/Itamarl/test",
12
  # 'mosaicml/mpt-7b-instruct',
 
7
  class EndpointHandler():
8
 
9
  def __init__(self, path=""):
10
+ path = 'mosaicml/mpt-7b'
11
  self.model = transformers.AutoModelForCausalLM.from_pretrained(path,
12
  #"/Users/itamarlevi/Downloads/my_repo_hf/hf/mpt-7b/venv/Itamarl/test",
13
  # 'mosaicml/mpt-7b-instruct',