RageshAntony commited on
Commit
2405833
·
verified ·
1 Parent(s): 9f16077

remove cache_dir=cache_dir

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -123,7 +123,7 @@ def load_pipeline(model_name):
123
  pipe = config["pipeline_class"].from_pretrained(
124
  config["repo_id"],
125
  torch_dtype=TORCH_DTYPE,
126
- cache_dir=cache_dir
127
  )
128
  pipe = pipe.to(DEVICE)
129
 
 
123
  pipe = config["pipeline_class"].from_pretrained(
124
  config["repo_id"],
125
  torch_dtype=TORCH_DTYPE,
126
+ #cache_dir=cache_dir
127
  )
128
  pipe = pipe.to(DEVICE)
129