shivansarora commited on
Commit
b5b1dda
·
verified ·
1 Parent(s): 04f3283
Files changed (1) hide show
  1. app.py +10 -0
app.py CHANGED
@@ -2,6 +2,16 @@ import gradio as gr
2
  import os
3
  import json
4
  import torch
 
 
 
 
 
 
 
 
 
 
5
  from parlai.core.opt import Opt
6
  from parlai.zoo.blender.blender_3B import download
7
  from parlai.core.agents import Agent
 
2
  import os
3
  import json
4
  import torch
5
+
6
+
7
+ _original_load = torch.load
8
+
9
+ def patched_load(*args, **kwargs):
10
+ file_path = args[0] if args else kwargs.get('f', 'UNKNOWN')
11
+ print(f"[DEBUG] torch.load is trying to load: {file_path}")
12
+ return _original_load(*args, **kwargs)
13
+
14
+ torch.load = patched_load
15
  from parlai.core.opt import Opt
16
  from parlai.zoo.blender.blender_3B import download
17
  from parlai.core.agents import Agent