amaresh8053 commited on
Commit
fb8a7a1
·
1 Parent(s): 58e7083

Update app.py

Browse files

disable weights-only mode

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -46,7 +46,7 @@ class Vocab:
46
  if not os.path.exists(CACHE_FILE):
47
  raise FileNotFoundError(f"Cache file {CACHE_FILE} not found. Please upload ubuntu_data_cache.pt")
48
 
49
- cache = torch.load(CACHE_FILE, map_location="cpu")
50
  train_pairs = cache["train"]
51
  test_pairs = cache["test"]
52
  vocab = cache["vocab"]
 
46
  if not os.path.exists(CACHE_FILE):
47
  raise FileNotFoundError(f"Cache file {CACHE_FILE} not found. Please upload ubuntu_data_cache.pt")
48
 
49
+ cache = torch.load(CACHE_FILE, map_location="cpu", weights_only=False)
50
  train_pairs = cache["train"]
51
  test_pairs = cache["test"]
52
  vocab = cache["vocab"]