Open models folder by default

This commit is contained in:
henk717 2022-02-21 00:46:18 +01:00
parent 7c678820cd
commit 300db651de
1 changed files with 1 additions and 2 deletions

View File

@ -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