Spaces:
Sleeping
Sleeping
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
|