Kowsher
/

TokenTrails

Text Generation

text-generation-inference

Model card Files Files and versions

Kowsher commited on Jul 18, 2023

Commit

c0158fa

·

1 Parent(s): dadc7b9

Update ChatFalcon.py

Files changed (1) hide show

ChatFalcon.py +1 -1

ChatFalcon.py CHANGED Viewed

@@ -36,7 +36,7 @@ from configuration_falcon import FalconConfig
 from transformers.models.auto import modeling_auto
 logger = logging.get_logger(__name__)
-modeling_auto.MODEL_FOR_CAUSAL_LM_MAPPING_NAMES['falcon'] = 'FalconForCausalLM'
 FALCON_PRETRAINED_MODEL_ARCHIVE_LIST = [
     "tiiuae/falcon-40b",
     "tiiuae/falcon-40b-instruct",

 from transformers.models.auto import modeling_auto
 logger = logging.get_logger(__name__)
 FALCON_PRETRAINED_MODEL_ARCHIVE_LIST = [
     "tiiuae/falcon-40b",
     "tiiuae/falcon-40b-instruct",