laragrl commited on
Commit
634283b
·
verified ·
1 Parent(s): ed5abde

Limit models to LeoLM due to ongoing access right requests

Browse files
Files changed (1) hide show
  1. app.py +7 -3
app.py CHANGED
@@ -3,10 +3,14 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
3
  import torch
4
 
5
  # Modelle und Tokenizer laden
 
 
 
 
 
 
6
  model_names = {
7
- "LeoLM_13B": "LeoLM/leo-hessianai-13b",
8
- "Occiglot_7B": "occiglot/occiglot-7b-de-en",
9
- "LLaMA2_13B": "meta-llama/Llama-2-13b-hf"
10
  }
11
 
12
  # Tokenizer und Modelle vorbereiten
 
3
  import torch
4
 
5
  # Modelle und Tokenizer laden
6
+ # model_names = {
7
+ # "LeoLM_13B": "LeoLM/leo-hessianai-13b",
8
+ # "Occiglot_7B": "occiglot/occiglot-7b-de-en",
9
+ # "LLaMA2_13B": "meta-llama/Llama-2-13b-hf"
10
+ # }
11
+
12
  model_names = {
13
+ "LeoLM_13B": "LeoLM/leo-hessianai-13b"
 
 
14
  }
15
 
16
  # Tokenizer und Modelle vorbereiten