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 = {}
|