dammy commited on
Commit
6588e48
·
1 Parent(s): 822a50d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
14
  import transformers
15
  import torch
16
 
17
- model_name = "tiiuae/falcon-40b-instruct"
18
 
19
  # model_name = 'google/flan-t5-base'
20
  # model = T5ForConditionalGeneration.from_pretrained(model_name, device_map='auto', offload_folder="offload")
 
14
  import transformers
15
  import torch
16
 
17
+ model = "tiiuae/falcon-40b-instruct"
18
 
19
  # model_name = 'google/flan-t5-base'
20
  # model = T5ForConditionalGeneration.from_pretrained(model_name, device_map='auto', offload_folder="offload")