Remove unused parts in Megatron-LM code and add copyright notice (#110)

2023-05-20 09:11:34 -06:00
parent b7955ef17b
commit 7297fa6f7c
8 changed files with 23 additions and 297 deletions
--- a/cacheflow/model_executor/parallel_utils/init.py
+++ b/cacheflow/model_executor/parallel_utils/init.py
@@ -1,6 +1,5 @@
 import cacheflow.model_executor.parallel_utils.parallel_state
 import cacheflow.model_executor.parallel_utils.tensor_parallel
-import cacheflow.model_executor.parallel_utils.utils

 # Alias parallel_state as mpu, its legacy name
 mpu = parallel_state
@@ -8,5 +7,4 @@ mpu = parallel_state
 __all__ = [
    "parallel_state",
    "tensor_parallel",
-    "utils",
 ]