Trim model path if using absolute path

This commit is contained in:
oobabooga 2023-09-19 13:13:24 -07:00
parent 869f47fff9
commit 9b7646140c

View File

@ -208,6 +208,7 @@ if __name__ == "__main__":
p = Path(shared.model_name) p = Path(shared.model_name)
if p.exists(): if p.exists():
model_name = p.parts[-1] model_name = p.parts[-1]
shared.model_name = model_name
else: else:
model_name = shared.model_name model_name = shared.model_name