Spaces:
Paused
Paused
Update utils.py
Browse files
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
|
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:
|