Spaces:
Running
on
Zero
Running
on
Zero
qinghuazhou
commited on
Commit
·
ec5e0f8
1
Parent(s):
fd0a592
use of cache in persistent storage
Browse files
app.py
CHANGED
@@ -11,7 +11,7 @@ from util import utils
|
|
11 |
|
12 |
## UTILITY FUNCTIONS ################################################
|
13 |
|
14 |
-
@spaces.GPU(duration=
|
15 |
def load_editor(model_name='gpt2-xl'):
|
16 |
|
17 |
# loading hyperparameters
|
@@ -23,7 +23,8 @@ def load_editor(model_name='gpt2-xl'):
|
|
23 |
hparams = hparams,
|
24 |
layer = 13,
|
25 |
edit_mode='in-place',
|
26 |
-
|
|
|
27 |
)
|
28 |
return editor
|
29 |
|
|
|
11 |
|
12 |
## UTILITY FUNCTIONS ################################################
|
13 |
|
14 |
+
@spaces.GPU(duration=180)
|
15 |
def load_editor(model_name='gpt2-xl'):
|
16 |
|
17 |
# loading hyperparameters
|
|
|
23 |
hparams = hparams,
|
24 |
layer = 13,
|
25 |
edit_mode='in-place',
|
26 |
+
cache_path='/data/cache/',
|
27 |
+
verbose=True
|
28 |
)
|
29 |
return editor
|
30 |
|