microsoft / lida

Automatic Generation of Visualizations and Infographics using Large Language Models
https://microsoft.github.io/lida/
MIT License
2.71k stars 290 forks source link

TypeError: Manager.__init__() got an unexpected keyword argument 'llm' , also the text_gen llm function does work correctly i had the change it #72

Open MohamedKHALILRouissi opened 10 months ago

MohamedKHALILRouissi commented 10 months ago
from lida import llm , Manager
from llmx.generators.text.hf_textgen import HFTextGenerator
text_gen = HFTextGenerator(provider="hf", models="uukuguy/speechless-llama2-hermes-orca-platypus-13b",device_map="auto")
print("success loading the model") # the falcon7b caused error out oom it need larger gpu allocation 
lida = Manager(llm=text_gen)

sumamry = lida.summarize("order.csv") # summirize the dataset
ankitshah009 commented 9 months ago

Receiving the same problem on the custom model loading from huggingface. Is there any update on this problem?