alexkueck commited on
Commit
ff0f378
·
1 Parent(s): 2aef2e9

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -83,7 +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_Blaize(base_model, load_8bit=True):
87
  if torch.cuda.is_available():
88
  device = "cuda"
89
  else:
 
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"
89
  else: