MohamedRashad commited on
Commit
5925564
·
1 Parent(s): 74d9892

Refactor model ID handling in app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -10
app.py CHANGED
@@ -8,19 +8,11 @@ models_available = [
8
  "MohamedRashad/Arabic-Orpo-Llama-3-8B-Instruct",
9
  "silma-ai/SILMA-9B-Instruct-v0.1.1",
10
  "inceptionai/jais-adapted-7b-chat",
11
- "inceptionai/jais-adapted-13b-chat",
12
  "inceptionai/jais-family-6p7b-chat",
13
- "inceptionai/jais-family-13b-chat",
14
  ]
15
 
16
- # Loop over all models to download them
17
- for model_id in models_available:
18
- try:
19
- AutoTokenizer.from_pretrained(model_id)
20
- AutoModelForCausalLM.from_pretrained(model_id)
21
- except:
22
- models_available.remove(model_id)
23
-
24
  tokenizer_a, model_a = None, None
25
  tokenizer_b, model_b = None, None
26
 
 
8
  "MohamedRashad/Arabic-Orpo-Llama-3-8B-Instruct",
9
  "silma-ai/SILMA-9B-Instruct-v0.1.1",
10
  "inceptionai/jais-adapted-7b-chat",
11
+ # "inceptionai/jais-adapted-13b-chat",
12
  "inceptionai/jais-family-6p7b-chat",
13
+ # "inceptionai/jais-family-13b-chat",
14
  ]
15
 
 
 
 
 
 
 
 
 
16
  tokenizer_a, model_a = None, None
17
  tokenizer_b, model_b = None, None
18