Балаганский Никита Николаевич
commited on
Commit
•
367e85c
1
Parent(s):
bf20d4a
fix
Browse files
app.py
CHANGED
|
@@ -104,7 +104,7 @@ def main():
|
|
| 104 |
|
| 105 |
|
| 106 |
|
| 107 |
-
@st.cache
|
| 108 |
def load_generator(lm_model_name: str) -> Generator:
|
| 109 |
with st.spinner('Loading language model...'):
|
| 110 |
generator = Generator(lm_model_name=lm_model_name, device=device)
|
|
|
|
| 104 |
|
| 105 |
|
| 106 |
|
| 107 |
+
@st.cache(hash_funcs={tokenizers.Tokenizer: lambda lm_tokenizer: hash(lm_tokenizer.to_str)}, allow_output_mutation=True)
|
| 108 |
def load_generator(lm_model_name: str) -> Generator:
|
| 109 |
with st.spinner('Loading language model...'):
|
| 110 |
generator = Generator(lm_model_name=lm_model_name, device=device)
|