m1k3wn commited on
Commit
5380536
·
verified ·
1 Parent(s): 310dbc9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -76,7 +76,7 @@ class ModelManager:
76
  tokenizer = T5Tokenizer.from_pretrained(
77
  model_path,
78
  token=HF_TOKEN,
79
- local_files_only=False
80
  )
81
  logger.debug("Tokenizer loaded successfully")
82
  except Exception as e:
@@ -88,7 +88,7 @@ class ModelManager:
88
  model = T5ForConditionalGeneration.from_pretrained(
89
  model_path,
90
  token=HF_TOKEN,
91
- local_files_only=False,
92
  low_cpu_mem_usage=True,
93
  torch_dtype=torch.float32
94
  )
 
76
  tokenizer = T5Tokenizer.from_pretrained(
77
  model_path,
78
  token=HF_TOKEN,
79
+ # local_files_only=False
80
  )
81
  logger.debug("Tokenizer loaded successfully")
82
  except Exception as e:
 
88
  model = T5ForConditionalGeneration.from_pretrained(
89
  model_path,
90
  token=HF_TOKEN,
91
+ # local_files_only=False,
92
  low_cpu_mem_usage=True,
93
  torch_dtype=torch.float32
94
  )