ThomasBlumet commited on
Commit
b0e2427
·
1 Parent(s): 8d9e0dc

change model

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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,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")
 
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")