Upload ModularStarEncoder
Browse files- modularStarEncoder.py +7 -1
modularStarEncoder.py
CHANGED
|
@@ -1,5 +1,11 @@
|
|
| 1 |
from transformers import AutoConfig, Starcoder2Model, Starcoder2Config
|
| 2 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 3 |
import math
|
| 4 |
import os
|
| 5 |
import warnings
|
|
|
|
| 1 |
from transformers import AutoConfig, Starcoder2Model, Starcoder2Config
|
| 2 |
+
import sys
|
| 3 |
+
import os
|
| 4 |
+
# Get the absolute path of the directory containing main.py
|
| 5 |
+
current_dir = os.path.dirname(os.path.abspath(__file__))
|
| 6 |
+
# Append this directory to sys.path
|
| 7 |
+
sys.path.insert(0, current_dir)
|
| 8 |
+
from config import ModularStarEncoderConfig
|
| 9 |
import math
|
| 10 |
import os
|
| 11 |
import warnings
|