alexkueck commited on
Commit
4aff23a
·
1 Parent(s): 87bd6bc

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -0
utils.py CHANGED
@@ -83,6 +83,7 @@ def load_tokenizer_and_model(base_model, load_8bit=False):
83
  return tokenizer,model, device
84
 
85
 
 
86
  def load_tokenizer_and_model_Baize(base_model, load_8bit=True):
87
  if torch.cuda.is_available():
88
  device = "cuda"
 
83
  return tokenizer,model, device
84
 
85
 
86
+ # hier werden aber Chat-Daten geladen!!!!
87
  def load_tokenizer_and_model_Baize(base_model, load_8bit=True):
88
  if torch.cuda.is_available():
89
  device = "cuda"