Spaces:
Running
on
Zero
Running
on
Zero
remove cache_dir=cache_dir
Browse files
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 |
|