Hugging Face's logo

Spaces:
abhattac
/
gpt-test-v2
Runtime error

App Files Community
abhattac commited on
Commit
0585818
•
1 Parent(s): d61d41e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
5
 
6
  os.environ["CURL_CA_BUNDLE"]=""
7
 
8
- tokenizer = AutoTokenizer.from_pretrained("togethercomputer/GPT-NeoXT-Chat-Base-20B", force_download=True)
9
  model = AutoModelForCausalLM.from_pretrained("togethercomputer/GPT-NeoXT-Chat-Base-20B", torch_dtype=torch.bfloat16)
10
 
11
  def gpt(prompt):
 
5
 
6
  os.environ["CURL_CA_BUNDLE"]=""
7
 
8
+ tokenizer = AutoTokenizer.from_pretrained("togethercomputer/GPT-NeoXT-Chat-Base-20B", force_download=True, local_files_only=False)
9
  model = AutoModelForCausalLM.from_pretrained("togethercomputer/GPT-NeoXT-Chat-Base-20B", torch_dtype=torch.bfloat16)
10
 
11
  def gpt(prompt):