search

YuxinWenRick / tree-ring-watermark

MIT License
250 stars 32 forks source link

Hi I want replicate your work but I think the datasets.py file is missing or do I need to create one myself? #5

Closed Xdarii closed 1 year ago

Xdarii commented 1 year ago

https://github.com/YuxinWenRick/tree-ring-watermark/blob/8ff17113d4138491dc330566bb1f1ad457c76ea0/optim_utils.py#L3

YuxinWenRick commented 1 year ago

Hi, sorry about the confusion. It's actually just huggingface datasets library. You can simply install it with pip install datasets, https://huggingface.co/docs/datasets/installation.