Spaces:
Runtime error
Runtime error
Commit
·
cab9610
1
Parent(s):
ab3c64e
Update app.py
Browse files
app.py
CHANGED
|
@@ -12,10 +12,9 @@ auth_token = os.environ.get("AUTH_TOKEN_SECRET")
|
|
| 12 |
|
| 13 |
tokenizer = LlamaTokenizer.from_pretrained("Claimed/capybara", use_auth_token=auth_token)
|
| 14 |
|
| 15 |
-
|
| 16 |
-
|
| 17 |
-
|
| 18 |
-
|
| 19 |
|
| 20 |
|
| 21 |
def add_text(history, text):
|
|
|
|
| 12 |
|
| 13 |
tokenizer = LlamaTokenizer.from_pretrained("Claimed/capybara", use_auth_token=auth_token)
|
| 14 |
|
| 15 |
+
model = LlamaForCausalLM.from_pretrained(
|
| 16 |
+
"Claimed/capybara", use_auth_token=auth_token)
|
| 17 |
+
model = model.to('cuda')
|
|
|
|
| 18 |
|
| 19 |
|
| 20 |
def add_text(history, text):
|