Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -6,7 +6,7 @@ import pickle
|
|
6 |
from transformers import AutoTokenizer, PreTrainedModel, PretrainedConfig
|
7 |
from huggingface_hub import login, hf_hub_download
|
8 |
import time
|
9 |
-
from
|
10 |
|
11 |
# Device setup
|
12 |
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|
|
|
6 |
from transformers import AutoTokenizer, PreTrainedModel, PretrainedConfig
|
7 |
from huggingface_hub import login, hf_hub_download
|
8 |
import time
|
9 |
+
from ch09util import subsequent_mask, create_model # Ensure ch09util.py is available
|
10 |
|
11 |
# Device setup
|
12 |
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|