Spaces:
Sleeping
Sleeping
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"
|