HowieHwong/TrustLLM

The parameter "device" of the function "LLMGeneration" can not work !

Closed this issue · 0 comments

xingyu8wang commented

image
The parameter "device" of the function "LLMGeneration" can not work !
Here is the Traceback:

Traceback (most recent call last):
File "/data/TrustLLM/run.py", line 3, in
llm_gen = LLMGeneration(
TypeError: init() got an unexpected keyword argument 'device'