Spaces:
Runtime error
Runtime error
Commit
·
088871e
1
Parent(s):
6d10c64
Update app.py
Browse files
app.py
CHANGED
|
@@ -8,10 +8,8 @@ import transformers
|
|
| 8 |
from transformers import AutoTokenizer, AutoConfig, LlamaForCausalLM, LlamaTokenizer, GenerationConfig
|
| 9 |
|
| 10 |
|
| 11 |
-
#auth_token = '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' #
|
| 12 |
auth_token = os.environ.get("AUTH_TOKEN_SECRET")
|
| 13 |
|
| 14 |
-
|
| 15 |
tokenizer = LlamaTokenizer.from_pretrained("Claimed/capybara", use_auth_token=auth_token)
|
| 16 |
|
| 17 |
# model = LlamaForCausalLM.from_pretrained(
|
|
|
|
| 8 |
from transformers import AutoTokenizer, AutoConfig, LlamaForCausalLM, LlamaTokenizer, GenerationConfig
|
| 9 |
|
| 10 |
|
|
|
|
| 11 |
auth_token = os.environ.get("AUTH_TOKEN_SECRET")
|
| 12 |
|
|
|
|
| 13 |
tokenizer = LlamaTokenizer.from_pretrained("Claimed/capybara", use_auth_token=auth_token)
|
| 14 |
|
| 15 |
# model = LlamaForCausalLM.from_pretrained(
|