malhajar commited on
Commit
8151a26
1 Parent(s): bf7b54b

Update scripts/create_request_file.py

Browse files
Files changed (1) hide show
  1. scripts/create_request_file.py +9 -8
scripts/create_request_file.py CHANGED
@@ -9,7 +9,7 @@ from colorama import Fore
9
  from huggingface_hub import HfApi, snapshot_download
10
 
11
  EVAL_REQUESTS_PATH = "eval-queue"
12
- QUEUE_REPO = "open-llm-leaderboard/requests"
13
 
14
  precisions = ("float16", "bfloat16", "8bit (LLM.int8)", "4bit (QLoRA / FP4)", "GPTQ")
15
  model_types = ("pretrained", "fine-tuned", "RL-tuned", "instruction-tuned")
@@ -38,13 +38,14 @@ def main():
38
  current_time = datetime.now(timezone.utc).strftime("%Y-%m-%dT%H:%M:%SZ")
39
  snapshot_download(repo_id=QUEUE_REPO, revision="main", local_dir=EVAL_REQUESTS_PATH, repo_type="dataset")
40
 
41
- model_name = click.prompt("Enter model name")
42
- revision = click.prompt("Enter revision", default="main")
43
- precision = click.prompt("Enter precision", default="float16", type=click.Choice(precisions))
44
- model_type = click.prompt("Enter model type", type=click.Choice(model_types))
45
- weight_type = click.prompt("Enter weight type", default="Original", type=click.Choice(weight_types))
46
- base_model = click.prompt("Enter base model", default="")
47
- status = click.prompt("Enter status", default="FINISHED")
 
48
 
49
  try:
50
  model_info = api.model_info(repo_id=model_name, revision=revision)
 
9
  from huggingface_hub import HfApi, snapshot_download
10
 
11
  EVAL_REQUESTS_PATH = "eval-queue"
12
+ QUEUE_REPO = "le-leadboard/requests"
13
 
14
  precisions = ("float16", "bfloat16", "8bit (LLM.int8)", "4bit (QLoRA / FP4)", "GPTQ")
15
  model_types = ("pretrained", "fine-tuned", "RL-tuned", "instruction-tuned")
 
38
  current_time = datetime.now(timezone.utc).strftime("%Y-%m-%dT%H:%M:%SZ")
39
  snapshot_download(repo_id=QUEUE_REPO, revision="main", local_dir=EVAL_REQUESTS_PATH, repo_type="dataset")
40
 
41
+ model_name = click.prompt("Entrez le nom du modèle")
42
+ revision = click.prompt("Entrez la révision", default="main")
43
+ precision = click.prompt("Entrez la précision", default="float16", type=click.Choice(precisions))
44
+ model_type = click.prompt("Entrez le type de modèle", type=click.Choice(model_types))
45
+ weight_type = click.prompt("Entrez le type de poids", default="Original", type=click.Choice(weight_types))
46
+ base_model = click.prompt("Entrez le modèle de base", default="")
47
+ status = click.prompt("Entrez le statut", default="FINISHED")
48
+
49
 
50
  try:
51
  model_info = api.model_info(repo_id=model_name, revision=revision)