hahahafofo commited on
Commit
0e8caab
1 Parent(s): 4fb6448
Files changed (1) hide show
  1. 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, device=0)
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