alexkueck commited on
Commit
7fe6015
·
1 Parent(s): 1e303bf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -80,13 +80,13 @@ login(token=os.environ["HF_ACCESS_TOKEN"]) #for read access!!!!
80
  #Modelle und Tokenizer
81
 
82
  #Alternativ mit beliebigen Modellen:
83
- #base_model = "project-baize/baize-v2-7b" #load_8bit = False (in load_tokenizer_and_model)
84
  #base_model = "TheBloke/airoboros-13B-HF" #load_8bit = False (in load_tokenizer_and_model)
85
  #base_model = "EleutherAI/gpt-neo-1.3B" #load_8bit = False (in load_tokenizer_and_model)
86
  #base_model = "TheBloke/airoboros-13B-HF" #load_8bit = True
87
  #base_model = "TheBloke/vicuna-13B-1.1-HF" #load_8bit = ?
88
  #base_model="gpt2-xl"
89
- base_model = "togethercomputer/LLaMA-2-7B-32K" #load_8bit = ? , trust_remote_code=True, torch_dtype=torch.float16
90
 
91
  # Load model directly
92
 
 
80
  #Modelle und Tokenizer
81
 
82
  #Alternativ mit beliebigen Modellen:
83
+ base_model = "project-baize/baize-v2-7b" #load_8bit = False (in load_tokenizer_and_model)
84
  #base_model = "TheBloke/airoboros-13B-HF" #load_8bit = False (in load_tokenizer_and_model)
85
  #base_model = "EleutherAI/gpt-neo-1.3B" #load_8bit = False (in load_tokenizer_and_model)
86
  #base_model = "TheBloke/airoboros-13B-HF" #load_8bit = True
87
  #base_model = "TheBloke/vicuna-13B-1.1-HF" #load_8bit = ?
88
  #base_model="gpt2-xl"
89
+
90
 
91
  # Load model directly
92