Spaces:
Running
Running
Update config.py
Browse files
config.py
CHANGED
@@ -4,18 +4,7 @@ from data_structures import ModelInfo
|
|
4 |
|
5 |
INITIAL_PEERS = PUBLIC_INITIAL_PEERS
|
6 |
|
7 |
-
MODELS = [
|
8 |
-
ModelInfo(
|
9 |
-
dht_prefix="StableBeluga2-hf",
|
10 |
-
repository="https://huggingface.co/petals-team/StableBeluga2",
|
11 |
-
num_blocks=80,
|
12 |
-
),
|
13 |
-
ModelInfo(
|
14 |
-
dht_prefix="falcon-180B-chat",
|
15 |
-
repository="https://huggingface.co/tiiuae/falcon-180B-chat",
|
16 |
-
num_blocks=80,
|
17 |
-
limited=True,
|
18 |
-
),
|
19 |
ModelInfo(
|
20 |
dht_prefix="Llama-2-70b-chat-hf",
|
21 |
repository="https://huggingface.co/meta-llama/Llama-2-70b-chat-hf",
|
@@ -26,6 +15,11 @@ MODELS = [
|
|
26 |
repository="https://huggingface.co/meta-llama/Llama-2-70b-hf",
|
27 |
num_blocks=80,
|
28 |
),
|
|
|
|
|
|
|
|
|
|
|
29 |
]
|
30 |
|
31 |
UPDATE_PERIOD = 60
|
|
|
4 |
|
5 |
INITIAL_PEERS = PUBLIC_INITIAL_PEERS
|
6 |
|
7 |
+
MODELS = [
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
8 |
ModelInfo(
|
9 |
dht_prefix="Llama-2-70b-chat-hf",
|
10 |
repository="https://huggingface.co/meta-llama/Llama-2-70b-chat-hf",
|
|
|
15 |
repository="https://huggingface.co/meta-llama/Llama-2-70b-hf",
|
16 |
num_blocks=80,
|
17 |
),
|
18 |
+
ModelInfo(
|
19 |
+
dht_prefix="beomi/llama-2-koen-13b",
|
20 |
+
repository="https://huggingface.co/beomi/llama-2-koen-13b",
|
21 |
+
num_blocks=80,
|
22 |
+
),
|
23 |
]
|
24 |
|
25 |
UPDATE_PERIOD = 60
|