Remove dependency

DBGGG · Mar 7, 2023 · f1afae7 · f1afae7
1 parent 616446d
commit f1afae7
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/codegeex/__init__.py b/codegeex/__init__.py
@@ -1,7 +1,6 @@
 import copy
 
 from typing import *
-from codegeex.megatron.model import CodeGeeXModel
 from codegeex.tokenizer import CodeGeeXTokenizer
 from codegeex.torch.inference import get_token_stream
 
@@ -14,7 +13,7 @@ def get_model(
 
 
 def generate(
-    model: CodeGeeXModel, 
+    model, 
     tokenizer: CodeGeeXTokenizer, 
     prompt: str, 
     out_seq_length: int,