mrmax14 commited on
Commit
330b518
·
1 Parent(s): 82eea23

new llama model added

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -18,7 +18,7 @@ repo_path = "meta-llama/Llama-2-7b-chat"
18
  params_path = os.path.join(repo_path, "params.json")
19
  model_weights_path = os.path.join(repo_path, "consolidated.00.pth")
20
  config_path = os.path.join(repo_path, "config.json")
21
- tokenizer_path = os.path.join(repo_path, "tokenizer.model")
22
 
23
  # Створюємо директорію для файлів, якщо вона не існує
24
  os.makedirs(repo_path, exist_ok=True)
@@ -41,7 +41,7 @@ if not os.path.exists(config_path):
41
  json.dump(custom_config, f, indent=4)
42
  print(f"{config_path} created successfully!")
43
 
44
- # Завантаження токенізатора
45
  tokenizer = AutoTokenizer.from_pretrained(tokenizer_path, use_auth_token=hf_token)
46
 
47
  # Завантаження стану моделі
 
18
  params_path = os.path.join(repo_path, "params.json")
19
  model_weights_path = os.path.join(repo_path, "consolidated.00.pth")
20
  config_path = os.path.join(repo_path, "config.json")
21
+ tokenizer_path = os.path.join(repo_path) # Вказуємо на сам репозиторій, а не окремий файл
22
 
23
  # Створюємо директорію для файлів, якщо вона не існує
24
  os.makedirs(repo_path, exist_ok=True)
 
41
  json.dump(custom_config, f, indent=4)
42
  print(f"{config_path} created successfully!")
43
 
44
+ # Завантаження токенізатора (вказуємо на весь репозиторій)
45
  tokenizer = AutoTokenizer.from_pretrained(tokenizer_path, use_auth_token=hf_token)
46
 
47
  # Завантаження стану моделі