Kowsher
/

TokenTrails

Text Generation

text-generation-inference

Model card Files Files and versions

Kowsher commited on Jul 18, 2023

Commit

9fec138

·

1 Parent(s): 8a94878

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -4,7 +4,7 @@ from typing import Any, Dict
 from transformers import AutoModelForCausalLM, AutoTokenizer
 from transformers.models.auto import modeling_auto
-modeling_auto.MODEL_FOR_CAUSAL_LM_MAPPING_NAMES['falcon'] = 'FalconForCausalLM'
 class EndpointHandler:
     def __init__(self, path=""):
         # load model and tokenizer from path

 from transformers import AutoModelForCausalLM, AutoTokenizer
 from transformers.models.auto import modeling_auto
 class EndpointHandler:
     def __init__(self, path=""):
         # load model and tokenizer from path