Question About Gustavosta/Stable-Diffusion-Prompts

[whybfq]

1. May I ask the dataset Gustavosta/Stable-Diffusion-Prompts: is it this link Gustavosta/Stable-Dif? I used train.parquet and convert it to jsonal while it seems incorrct, thanks for your reply. :)
   so when I run the code det_mem_viz.ipynb, I met the error:

IndexError Traceback (most recent call last)
in <cell line: 11>()
21
22 curr_data = np.array(row[key])
---> 23 curr_data = curr_data[:num_gens, start:end]
24 curr_data = np.mean(curr_data, axis=0)
25

IndexError: too many indices for array: array is 0-dimensional, but 2 were indexed

I think maybe I reproduced the dataset non_memorized_prompts incorrect,

[whybfq]

I have found the error, thanks.