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