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

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
  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",
 
7
  class EndpointHandler():
8
 
9
  def __init__(self, path=""):
10
+ print(f"Hugging face handler path {path}")
11
  path = 'mosaicml/mpt-7b'
12
  self.model = transformers.AutoModelForCausalLM.from_pretrained(path,
13
  #"/Users/itamarlevi/Downloads/my_repo_hf/hf/mpt-7b/venv/Itamarl/test",