Kowsher commited on
Commit
c0158fa
1 Parent(s): dadc7b9

Update ChatFalcon.py

Browse files
Files changed (1) hide show
  1. ChatFalcon.py +1 -1
ChatFalcon.py CHANGED
@@ -36,7 +36,7 @@ from configuration_falcon import FalconConfig
36
  from transformers.models.auto import modeling_auto
37
 
38
  logger = logging.get_logger(__name__)
39
- modeling_auto.MODEL_FOR_CAUSAL_LM_MAPPING_NAMES['falcon'] = 'FalconForCausalLM'
40
  FALCON_PRETRAINED_MODEL_ARCHIVE_LIST = [
41
  "tiiuae/falcon-40b",
42
  "tiiuae/falcon-40b-instruct",
 
36
  from transformers.models.auto import modeling_auto
37
 
38
  logger = logging.get_logger(__name__)
39
+
40
  FALCON_PRETRAINED_MODEL_ARCHIVE_LIST = [
41
  "tiiuae/falcon-40b",
42
  "tiiuae/falcon-40b-instruct",