Harley-ml commited on
Commit
ab91e93
·
verified ·
1 Parent(s): 9acb03b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -17
app.py CHANGED
@@ -1,20 +1,3 @@
1
- """Gradio Space for Harley-ml text-generation.
2
-
3
- It exposes:
4
- - prompt
5
- - model selection
6
- - temperature
7
- - top_k
8
- - top_p
9
- - repetition penalty
10
- - max_new_tokens
11
- - do_sample
12
- - return_full_text
13
-
14
- The app loads and caches each model so you can switch between
15
- different models without re-downloading on every request.
16
- """
17
-
18
  from __future__ import annotations
19
 
20
  import os
@@ -43,6 +26,7 @@ MODEL_CHOICES: List[Tuple[str, str]] = [
43
  ("LWTDense", "Harley-ml/LWTDense-6M"),
44
  ("MiniMD-28M", "Harley-ml/MiniMD-28M"),
45
  ("StopAskingQuestionsMini-656k", "Harley-ml/StopAskingQuestionsMini-656k"),
 
46
  ]
47
 
48
  LABEL_TO_REPO: Dict[str, str] = {label: repo for label, repo in MODEL_CHOICES}
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  from __future__ import annotations
2
 
3
  import os
 
26
  ("LWTDense", "Harley-ml/LWTDense-6M"),
27
  ("MiniMD-28M", "Harley-ml/MiniMD-28M"),
28
  ("StopAskingQuestionsMini-656k", "Harley-ml/StopAskingQuestionsMini-656k"),
29
+ ("Dillion-1.2M", "Harley-ml/Dillion-1.2M")
30
  ]
31
 
32
  LABEL_TO_REPO: Dict[str, str] = {label: repo for label, repo in MODEL_CHOICES}