IC4T commited on
Commit
328524f
·
1 Parent(s): 243abbf
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -18,6 +18,7 @@ from langchain.llms import HuggingFacePipeline
18
  from instruct_pipeline import InstructionTextGenerationPipeline
19
 
20
  from training.generate import load_model_tokenizer_for_generate
 
21
 
22
  # from training.generate import InstructionTextGenerationPipeline, load_model_tokenizer_for_generate
23
  # from googletrans import Translator
 
18
  from instruct_pipeline import InstructionTextGenerationPipeline
19
 
20
  from training.generate import load_model_tokenizer_for_generate
21
+ from ctransformers import AutoModelForCausalLM
22
 
23
  # from training.generate import InstructionTextGenerationPipeline, load_model_tokenizer_for_generate
24
  # from googletrans import Translator