I am running the code found on this [site][1] in my server and I would like to store images instead of showing them since I have connected remotely with an ssh connection to my server via an SSH connection. The code is for instance this one: skip_tokens = [1] # skip the special token for the start of the text <s> inp = TextTokenInput( eval_prompt, tokenizer, skip_tokens=skip_tokens, ) target = "playing guitar, hiking, and spending time with his family." attr_res = llm_attr.attribute(inp, target=target, skip_tokens=skip_tokens) attr_res.plot_token_attr(show=True) How to store the files locally instead of showing them? [1]: https://captum.ai/tutorials/Llama2_LLM_Attribution Continue reading...
