Spaces:

rodrigomasini
/

rephrase

Paused

rodrigomasini commited on Nov 8, 2023

Commit

0831088

1 Parent(s): 4ab4748

Update app_v4.py

Files changed (1) hide show

app_v4.py CHANGED Viewed

@@ -30,7 +30,7 @@ if device == "cuda:0":
     torch.cuda.empty_cache()
 # Load tokenizer
-tokenizer = AutoTokenizer.from_pretrained(pretrained_model_dir, use_fast=True)
 # Attempt to load the model, catch any OOM errors
 try:

     torch.cuda.empty_cache()
 # Load tokenizer
+tokenizer = AutoTokenizer.from_pretrained(pretrained_model_dir, use_fast=False)
 # Attempt to load the model, catch any OOM errors
 try: