5
0
justheuristic 3 жил өмнө
parent
commit
de556c99be

+ 1 - 0
cli/convert_model.py

@@ -69,6 +69,7 @@ if __name__ == "__main__":
         f" - {args.output_repo}@{args.block_branch_prefix}{len(transformer_blocks)}"
     )
     for i, block in enumerate(tqdm(transformer_blocks)):
+        repo.git_checkout(args.client_branch, create_branch_ok=True)
         with repo.commit(
             commit_message=args.commit_message, branch=args.block_branch_prefix + str(i), track_large_files=True
         ):

+ 3 - 3
src/bloom/__init__.py

@@ -1,3 +1,3 @@
-from src.bloom import *
-from src.client import *
-from src.server import *
+from src.bloom.block import BloomBlock
+from src.bloom.model import BloomConfig, BloomModel, BloomPreTrainedModel, BloomForCausalLM
+from src.bloom.block import BloomBlock