Open models folder by default
This commit is contained in:
parent
7c678820cd
commit
300db651de
|
@ -236,8 +236,7 @@ def getModelSelection():
|
||||||
# If custom model was selected, get the filesystem location and store it
|
# If custom model was selected, get the filesystem location and store it
|
||||||
if(vars.model == "NeoCustom" or vars.model == "GPT2Custom"):
|
if(vars.model == "NeoCustom" or vars.model == "GPT2Custom"):
|
||||||
print("{0}Please choose the folder where pytorch_model.bin is located:{1}\n".format(colors.CYAN, colors.END))
|
print("{0}Please choose the folder where pytorch_model.bin is located:{1}\n".format(colors.CYAN, colors.END))
|
||||||
|
modpath = fileops.getdirpath(getcwd() + "/models", "Select Model Folder")
|
||||||
modpath = fileops.getdirpath(getcwd(), "Select Model Folder")
|
|
||||||
|
|
||||||
if(modpath):
|
if(modpath):
|
||||||
# Save directory to vars
|
# Save directory to vars
|
||||||
|
|
Loading…
Reference in New Issue