Spaces:
Sleeping
Sleeping
IC4T
commited on
Commit
·
06f0841
1
Parent(s):
177b631
commit
Browse files
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(
|