mirror of
https://github.com/nod-ai/AMD-SHARK-Studio.git
synced 2026-02-19 11:56:43 -05:00
Fix bloom lint
This commit is contained in:
@@ -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()))
|
||||
|
||||
@@ -378,4 +378,4 @@ if __name__ == "__main__":
|
||||
|
||||
f = open(f"{working_dir}/prompt.txt", "w+")
|
||||
f.write(prompt + next_token)
|
||||
f.close()
|
||||
f.close()
|
||||
|
||||
Reference in New Issue
Block a user