Kowsher commited on
Commit
9fec138
1 Parent(s): 8a94878

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -4,7 +4,7 @@ from typing import Any, Dict
4
  from transformers import AutoModelForCausalLM, AutoTokenizer
5
  from transformers.models.auto import modeling_auto
6
 
7
- modeling_auto.MODEL_FOR_CAUSAL_LM_MAPPING_NAMES['falcon'] = 'FalconForCausalLM'
8
  class EndpointHandler:
9
  def __init__(self, path=""):
10
  # load model and tokenizer from path
 
4
  from transformers import AutoModelForCausalLM, AutoTokenizer
5
  from transformers.models.auto import modeling_auto
6
 
7
+
8
  class EndpointHandler:
9
  def __init__(self, path=""):
10
  # load model and tokenizer from path