Hugging Face's logo

Spaces:
msy127
/
TheBloke-Llama-2-7b-Chat-GPTQ
Runtime error

App Files Community
msy127 commited on
Commit
de28914
·
1 Parent(s): e424c6f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -1,6 +1,9 @@
1
  import torch
2
  from transformers import AutoTokenizer, AutoModelForCausalLM
3
 
 
 
 
4
  tokenizer = AutoTokenizer.from_pretrained("NousResearch/Llama-2-7b-chat-hf")
5
  model = AutoModelForCausalLM.from_pretrained("NousResearch/Llama-2-7b-chat-hf")
6
 
 
1
  import torch
2
  from transformers import AutoTokenizer, AutoModelForCausalLM
3
 
4
+ model_name = "facebook/opt-350m"
5
+ # model_name = "NousResearch/Llama-2-7b-chat-hf"
6
+
7
  tokenizer = AutoTokenizer.from_pretrained("NousResearch/Llama-2-7b-chat-hf")
8
  model = AutoModelForCausalLM.from_pretrained("NousResearch/Llama-2-7b-chat-hf")
9