amiguel commited on
Commit
f82672b
·
verified ·
1 Parent(s): 4a47453

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 utils.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"
 
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"