ChaosTong commited on
Commit
a4f5c1e
·
1 Parent(s): 7929161
Files changed (2) hide show
  1. inference.py +3 -2
  2. output/out.jpg +0 -0
inference.py CHANGED
@@ -40,8 +40,9 @@ def profile(func):
40
 
41
 
42
  def auto_load_weight(weight, version=None, map_location=None):
43
- """Auto load Generator version from weight."""
44
- cache_dir = os.path.expanduser(".cache")
 
45
  weight_name = os.path.basename(weight).lower()
46
  cached_weight = os.path.join(cache_dir, weight_name)
47
 
 
40
 
41
 
42
  def auto_load_weight(weight, version=None, map_location=None):
43
+ """Auto load Generator version from weight."""
44
+ project_dir = os.path.dirname(os.path.abspath(__file__))
45
+ cache_dir = os.path.join(project_dir, ".cache")
46
  weight_name = os.path.basename(weight).lower()
47
  cached_weight = os.path.join(cache_dir, weight_name)
48
 
output/out.jpg CHANGED