suayptalha
commited on
Update modeling_minGRULM.py
Browse files- modeling_minGRULM.py +1 -0
modeling_minGRULM.py
CHANGED
@@ -137,6 +137,7 @@ class MinGRULMForCausalLM(PreTrainedModel):
|
|
137 |
save_directory (str): Directory to save the model.
|
138 |
safe_serialization (bool, optional): Whether to use safe serialization. Defaults to True.
|
139 |
"""
|
|
|
140 |
os.makedirs(save_directory, exist_ok=True)
|
141 |
|
142 |
if safe_serialization:
|
|
|
137 |
save_directory (str): Directory to save the model.
|
138 |
safe_serialization (bool, optional): Whether to use safe serialization. Defaults to True.
|
139 |
"""
|
140 |
+
import os
|
141 |
os.makedirs(save_directory, exist_ok=True)
|
142 |
|
143 |
if safe_serialization:
|