Update app.py
Browse files
app.py
CHANGED
@@ -14,7 +14,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
|
|
14 |
import transformers
|
15 |
import torch
|
16 |
|
17 |
-
|
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")
|