Fix bloom lint

This commit is contained in:
Anush Elangovan
2023-03-10 11:52:40 -08:00
parent b8f4b18951
commit ce6d82eab2
2 changed files with 2 additions and 4 deletions

View File

@@ -731,7 +731,6 @@ if __name__ == "__main__":
"Warning: If you need to use memory efficient mode, you probably want to use 'download' instead"
)
if not os.path.isdir(args.model_path):
os.mkdir(args.model_path)
@@ -760,7 +759,6 @@ if __name__ == "__main__":
config = json.load(f)
f.close()
self_path = os.path.dirname(os.path.abspath(__file__))
script_path = os.path.join(self_path, "sharded_bloom_large_models.py")
@@ -841,4 +839,4 @@ if __name__ == "__main__":
[input_ids, next_token.unsqueeze(-1)], dim=-1
)
print(tokenizer.decode(input_ids.squeeze()))
print(tokenizer.decode(input_ids.squeeze()))

View File

@@ -378,4 +378,4 @@ if __name__ == "__main__":
f = open(f"{working_dir}/prompt.txt", "w+")
f.write(prompt + next_token)
f.close()
f.close()