IC4T commited on
Commit
06f0841
·
1 Parent(s): 177b631
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -68,7 +68,8 @@ match model_type:
68
 
69
  # llm = AutoModelForCausalLM.from_pretrained(model_path, model_type='dolly-v2')
70
 
71
- llm = CTransformers(model_path, callbacks=[StreamingStdOutCallbackHandler()])
 
72
 
73
  # llm = HuggingFacePipeline(
74
  # pipeline=InstructionTextGenerationPipeline(
 
68
 
69
  # llm = AutoModelForCausalLM.from_pretrained(model_path, model_type='dolly-v2')
70
 
71
+ # llm = CTransformers(model_path, callbacks=[StreamingStdOutCallbackHandler()])
72
+ llm = CTransformers(model=model_path, model_type="dolly-v2", callbacks=[StreamingStdOutCallbackHandler()])
73
 
74
  # llm = HuggingFacePipeline(
75
  # pipeline=InstructionTextGenerationPipeline(