xinchen9 commited on
Commit
8ce9703
1 Parent(s): 7980d67

[Update]Add about and requirement files

Browse files
Files changed (2) hide show
  1. about.py +85 -0
  2. requirements.txt +14 -0
about.py ADDED
@@ -0,0 +1,85 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from dataclasses import dataclass
2
+ from enum import Enum
3
+
4
+ @dataclass
5
+ class Task:
6
+ benchmark: str
7
+ metric: str
8
+ col_name: str
9
+
10
+
11
+ # Select your tasks here
12
+ # ---------------------------------------------------
13
+ class Tasks(Enum):
14
+ # task_key in the json file, metric_key in the json file, name to display in the leaderboard
15
+ task0 = Task("anli_r1", "acc", "ANLI")
16
+ task1 = Task("logiqa", "acc_norm", "LogiQA")
17
+
18
+ NUM_FEWSHOT = 0 # Change with your few shot
19
+ # ---------------------------------------------------
20
+
21
+
22
+
23
+ # Your leaderboard name
24
+ TITLE = """<h1 align="center" id="space-title">UnlearnDiffAtk Benchmark</h1>"""
25
+
26
+ # subtitle
27
+ SUB_TITLE = """<h2 align="center" id="space-title">Effective and efficient adversarial prompt generation approach for diffusion models</h1>"""
28
+
29
+ # What does your leaderboard evaluate?
30
+ INTRODUCTION_TEXT = """
31
+ This benchmark is evaluates the robustness of safety-driven unlearned diffusion models (DMs)
32
+ (i.e., DMs after unlearning undesirable concepts, styles, or objects) across a variety of tasks. For more details, please visit the [project](https://www.optml-group.com/posts/mu_attack),
33
+ check the [code](https://github.com/OPTML-Group/Diffusion-MU-Attack), and read the [paper](https://arxiv.org/abs/2310.11868).\\
34
+ Demo of our offensive method: [UnlearnDiffAtk](https://huggingface.co/spaces/xinchen9/SD_Offense)\\
35
+ Demo of our defensive method: [AdvUnlearn](https://huggingface.co/spaces/xinchen9/SD_Defense)
36
+ """
37
+
38
+ # Which evaluations are you running? how can people reproduce what you have?
39
+ LLM_BENCHMARKS_TEXT = f"""
40
+ ## How it works
41
+
42
+ ## Reproducibility
43
+ To reproduce our results, here is the commands you can run:
44
+
45
+ """
46
+
47
+ EVALUATION_QUEUE_TEXT = """
48
+ ## Some good practices before submitting a model
49
+
50
+ ### 1) Make sure you can load your model and tokenizer using AutoClasses:
51
+ ```python
52
+ from transformers import AutoConfig, AutoModel, AutoTokenizer
53
+ config = AutoConfig.from_pretrained("your model name", revision=revision)
54
+ model = AutoModel.from_pretrained("your model name", revision=revision)
55
+ tokenizer = AutoTokenizer.from_pretrained("your model name", revision=revision)
56
+ ```
57
+ If this step fails, follow the error messages to debug your model before submitting it. It's likely your model has been improperly uploaded.
58
+
59
+ Note: make sure your model is public!
60
+ Note: if your model needs `use_remote_code=True`, we do not support this option yet but we are working on adding it, stay posted!
61
+
62
+ ### 2) Convert your model weights to [safetensors](https://huggingface.co/docs/safetensors/index)
63
+ It's a new format for storing weights which is safer and faster to load and use. It will also allow us to add the number of parameters of your model to the `Extended Viewer`!
64
+
65
+ ### 3) Make sure your model has an open license!
66
+ This is a leaderboard for Open LLMs, and we'd love for as many people as possible to know they can use your model 🤗
67
+
68
+ ### 4) Fill up your model card
69
+ When we add extra information about models to the leaderboard, it will be automatically taken from the model card
70
+
71
+ ## In case of model failure
72
+ If your model is displayed in the `FAILED` category, its execution stopped.
73
+ Make sure you have followed the above steps first.
74
+ If everything is done, check you can launch the EleutherAIHarness on your model locally, using the above command without modifications (you can add `--limit` to limit the number of examples per task).
75
+ """
76
+
77
+ CITATION_BUTTON_LABEL = "Copy the following snippet to cite these results"
78
+ CITATION_BUTTON_TEXT = r"""
79
+ @article{zhang2023generate,
80
+ title={To Generate or Not? Safety-Driven Unlearned Diffusion Models Are Still Easy To Generate Unsafe Images... For Now},
81
+ author={Zhang, Yimeng and Jia, Jinghan and Chen, Xin and Chen, Aochuan and Zhang, Yihua and Liu, Jiancheng and Ding, Ke and Liu, Sijia},
82
+ journal={arXiv preprint arXiv:2310.11868},
83
+ year={2023}
84
+ }
85
+ """
requirements.txt ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ APScheduler
2
+ black
3
+ click
4
+ datasets
5
+ gradio
6
+ gradio_client
7
+ huggingface-hub>=0.18.0
8
+ matplotlib
9
+ numpy
10
+ pandas
11
+ python-dateutil
12
+ requests
13
+ tqdm
14
+ transformers