fix glob path in hypernetwork.py
This commit is contained in:
@@ -43,7 +43,7 @@ class Hypernetwork:
|
|||||||
def load_hypernetworks(path):
|
def load_hypernetworks(path):
|
||||||
res = {}
|
res = {}
|
||||||
|
|
||||||
for filename in glob.iglob(path + '**/*.pt', recursive=True):
|
for filename in glob.iglob(os.path.join(path, '**/*.pt'), recursive=True):
|
||||||
try:
|
try:
|
||||||
hn = Hypernetwork(filename)
|
hn = Hypernetwork(filename)
|
||||||
res[hn.name] = hn
|
res[hn.name] = hn
|
||||||
|
|||||||
Reference in New Issue
Block a user