kevinwang676 commited on
Commit
80b2033
·
verified ·
1 Parent(s): deca852

Update api_v2.py

Browse files
Files changed (1) hide show
  1. api_v2.py +1 -0
api_v2.py CHANGED
@@ -101,6 +101,7 @@ import os
101
  import sys
102
  import traceback
103
  from typing import Generator
 
104
 
105
  now_dir = os.getcwd()
106
  sys.path.append(now_dir)
 
101
  import sys
102
  import traceback
103
  from typing import Generator
104
+ import torch
105
 
106
  now_dir = os.getcwd()
107
  sys.path.append(now_dir)