This commit is contained in:
xingjun.wang
2023-10-22 23:23:07 +08:00
parent f98e65a8c3
commit 9be3cac470

View File

@@ -64,6 +64,7 @@ class LLMPipeline(Pipeline):
self.device_map = 'cuda'
self.torch_dtype = kwargs.pop('torch_dtype', None)
self.ignore_file_pattern = kwargs.pop('ignore_file_pattern', None)
with self._temp_configuration_file(kwargs):
super().__init__(*args, **kwargs)