rodrigomasini commited on
Commit
72fd58a
1 Parent(s): ad7103c

Update app_v3.py

Browse files
Files changed (1) hide show
  1. app_v3.py +2 -2
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)