ThomasBlumet
commited on
Commit
·
b0e2427
1
Parent(s):
8d9e0dc
change model
Browse files
app.py
CHANGED
@@ -8,8 +8,8 @@ logger = logging.get_logger("transformers")
|
|
8 |
|
9 |
# Load the model and tokenizer
|
10 |
model_name = "openai-community/gpt2" #"openai-community/gpt2" or "TheBloke/Mistral-7B-Instruct-v0.1-GPTQ" or "TheBloke/Llama-2-7B-Chat-GGML" or "TheBloke/zephyr-7B-beta-GPTQ"
|
11 |
-
tokenizer = AutoTokenizer.from_pretrained(model_name
|
12 |
-
model = AutoModelForCausalLM.from_pretrained(model_name
|
13 |
|
14 |
#transfer model on GPU
|
15 |
#model.to("cuda")
|
|
|
8 |
|
9 |
# Load the model and tokenizer
|
10 |
model_name = "openai-community/gpt2" #"openai-community/gpt2" or "TheBloke/Mistral-7B-Instruct-v0.1-GPTQ" or "TheBloke/Llama-2-7B-Chat-GGML" or "TheBloke/zephyr-7B-beta-GPTQ"
|
11 |
+
tokenizer = AutoTokenizer.from_pretrained(model_name)#,use_fast=True
|
12 |
+
model = AutoModelForCausalLM.from_pretrained(model_name)#,device_map="auto",trust_remote_code=False,revision="main")
|
13 |
|
14 |
#transfer model on GPU
|
15 |
#model.to("cuda")
|