ldhldh commited on
Commit
c0f28eb
1 Parent(s): f41b280

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +6 -12
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