Update modeling_esm_plusplus.py
Browse files- modeling_esm_plusplus.py +1 -1
modeling_esm_plusplus.py
CHANGED
|
@@ -586,7 +586,7 @@ class ESMplusplusForMaskedLM(PreTrainedModel):
|
|
| 586 |
conn.commit()
|
| 587 |
|
| 588 |
conn.commit()
|
| 589 |
-
|
| 590 |
return None
|
| 591 |
|
| 592 |
embeddings_dict = {}
|
|
|
|
| 586 |
conn.commit()
|
| 587 |
|
| 588 |
conn.commit()
|
| 589 |
+
conn.close()
|
| 590 |
return None
|
| 591 |
|
| 592 |
embeddings_dict = {}
|