alexkueck commited on
Commit
491f375
·
1 Parent(s): bde0193

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -67,7 +67,7 @@ def generate_prompt_with_history(text, history, tokenizer, max_length=2048):
67
 
68
 
69
 
70
- def load_tokenizer_and_model(base_model,load_8bit=False, use_fast=False):
71
  if torch.cuda.is_available():
72
  device = "cuda"
73
  else:
 
67
 
68
 
69
 
70
+ def load_tokenizer_and_model(base_model,load_8bit=False):
71
  if torch.cuda.is_available():
72
  device = "cuda"
73
  else: