Remove dependency

2 years ago · f1afae7b26
parent 616446dc72
commit f1afae7b26
1 changed files with 1 additions and 2 deletions
--- 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,