Spaces:
Runtime error
Runtime error
Tao Wu commited on
Commit ·
a70f9b1
1
Parent(s): 5f285e1
init
Browse files- app/embedding_setup.py +1 -1
app/embedding_setup.py
CHANGED
|
@@ -9,7 +9,7 @@ import os
|
|
| 9 |
import torch
|
| 10 |
from predibase import Predibase, FinetuningConfig, DeploymentConfig
|
| 11 |
pb_auth = os.environ.get("pb_token")
|
| 12 |
-
pb = Predibase(api_token=
|
| 13 |
|
| 14 |
if torch.cuda.is_available():
|
| 15 |
device = "cuda"
|
|
|
|
| 9 |
import torch
|
| 10 |
from predibase import Predibase, FinetuningConfig, DeploymentConfig
|
| 11 |
pb_auth = os.environ.get("pb_token")
|
| 12 |
+
pb = Predibase(api_token='pb_8PhUywZ5VQCQFmGl8ZgfiQ')
|
| 13 |
|
| 14 |
if torch.cuda.is_available():
|
| 15 |
device = "cuda"
|