thepolymerguy commited on
Commit
4d15dc7
1 Parent(s): 14c7070

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -6,10 +6,13 @@ import torch.nn as nn
6
  import transformers
7
  from transformers import AutoTokenizer, AutoConfig, LlamaForCausalLM, LlamaTokenizer, GenerationConfig
8
 
9
- tokenizer = LlamaTokenizer.from_pretrained("Claimed/capybara")
 
 
 
10
 
11
  model = LlamaForCausalLM.from_pretrained(
12
- "Claimed/capybara")
13
  #model = model.to('cuda')
14
 
15
 
 
6
  import transformers
7
  from transformers import AutoTokenizer, AutoConfig, LlamaForCausalLM, LlamaTokenizer, GenerationConfig
8
 
9
+
10
+ auth_token = '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' #= os.environ.get("TOKEN_FROM_SECRET")
11
+
12
+ tokenizer = LlamaTokenizer.from_pretrained("Claimed/capybara", use_auth_token=auth_token)
13
 
14
  model = LlamaForCausalLM.from_pretrained(
15
+ "Claimed/capybara", use_auth_token=auth_token)
16
  #model = model.to('cuda')
17
 
18