Spaces:
Paused
Paused
Commit
·
6c491b0
1
Parent(s):
c950661
Update app_v3.py
Browse files
app_v3.py
CHANGED
@@ -33,8 +33,8 @@ model_basename = "Jackson2-4bit-128g-GPTQ"
|
|
33 |
|
34 |
use_triton = False
|
35 |
|
36 |
-
if torch.cuda.is_available():
|
37 |
-
torch.cuda.empty_cache()
|
38 |
|
39 |
#tokenizer = AutoTokenizer.from_pretrained(model_name_or_path, use_fast=True, legacy=False)
|
40 |
tokenizer = AutoTokenizer.from_pretrained(pretrained_model_dir, use_fast=True)
|
|
|
33 |
|
34 |
use_triton = False
|
35 |
|
36 |
+
#if torch.cuda.is_available():
|
37 |
+
# torch.cuda.empty_cache()
|
38 |
|
39 |
#tokenizer = AutoTokenizer.from_pretrained(model_name_or_path, use_fast=True, legacy=False)
|
40 |
tokenizer = AutoTokenizer.from_pretrained(pretrained_model_dir, use_fast=True)
|