Spaces:
Runtime error
Runtime error
hahahafofo
commited on
Commit
•
0e8caab
1
Parent(s):
4fb6448
fix bug
Browse files- chatpdf.py +1 -1
chatpdf.py
CHANGED
@@ -39,7 +39,7 @@ class ChatPDF:
|
|
39 |
elif gen_model_type == "llama":
|
40 |
self.gen_model = LlamaModel(gen_model_type, gen_model_name_or_path, lora_name=lora_model_name_or_path)
|
41 |
elif gen_model_type == "t5":
|
42 |
-
self.gen_model = pipeline('text2text-generation', model=gen_model_name_or_path
|
43 |
else:
|
44 |
raise ValueError('gen_model_type must be chatglm or llama.')
|
45 |
self.history = None
|
|
|
39 |
elif gen_model_type == "llama":
|
40 |
self.gen_model = LlamaModel(gen_model_type, gen_model_name_or_path, lora_name=lora_model_name_or_path)
|
41 |
elif gen_model_type == "t5":
|
42 |
+
self.gen_model = pipeline('text2text-generation', model=gen_model_name_or_path)
|
43 |
else:
|
44 |
raise ValueError('gen_model_type must be chatglm or llama.')
|
45 |
self.history = None
|