gba16326553 commited on
Commit
8ddffec
1 Parent(s): 0623c5b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ modelName = "bartowski/Mistral-Nemo-Instruct-2407-GGUF"
16
  modelId = "Mistral-Nemo-Instruct-2407-Q2_K.gguf"
17
  tokenizer = AutoTokenizer.from_pretrained(modelName,gguf_file=modelId)
18
 
19
- model = AutoModel.from_pretrained(modelName,gguf_file=modelId)
20
  #model = AutoModelForCausalLM.from_pretrained("google/gemma-2-2b-it", torch_dtype=torch.float16 )
21
  #stvlynn/Gemma-2-2b-Chinese-it
22
  #tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-large")
 
16
  modelId = "Mistral-Nemo-Instruct-2407-Q2_K.gguf"
17
  tokenizer = AutoTokenizer.from_pretrained(modelName,gguf_file=modelId)
18
 
19
+ model = AutoModel.from_pretrained(modelName,gguf_file=modelId,torch_dtype=torch.float16)
20
  #model = AutoModelForCausalLM.from_pretrained("google/gemma-2-2b-it", torch_dtype=torch.float16 )
21
  #stvlynn/Gemma-2-2b-Chinese-it
22
  #tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-large")