amiguel commited on
Commit
67aeb8b
·
verified ·
1 Parent(s): 9a4c5ac

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ import os
7
  from transformers import AutoTokenizer
8
  from huggingface_hub import login
9
  import time
10
- from utils.ch09util import subsequent_mask # Ensure ch09util.py is available
11
 
12
  # Device setup
13
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 
7
  from transformers import AutoTokenizer
8
  from huggingface_hub import login
9
  import time
10
+ from ch09util import subsequent_mask # Ensure ch09util.py is available
11
 
12
  # Device setup
13
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"