Spaces:
Runtime error
Runtime error
complete-dope
commited on
Commit
•
17ae86e
1
Parent(s):
1fd6029
dockerSecond
Browse files- main.py +12 -5
- maintain.py +2 -1
main.py
CHANGED
@@ -1,3 +1,8 @@
|
|
|
|
|
|
|
|
|
|
|
|
1 |
#this repo contains the code for mixtral model for finding the icd-10 codes and this scripts runs well on the single GPU and is now trying to run with the multiple GPU and i need to make sure that this script runs in a multi gpu environment
|
2 |
|
3 |
import warnings
|
@@ -147,9 +152,11 @@ trainer.train()
|
|
147 |
|
148 |
# final model prepared
|
149 |
'''
|
150 |
-
1) Make sure the model runs on multi gpu script !
|
151 |
-
2) The dataset is loaded
|
152 |
-
3) The langchain implementation to oversee the prompt generation guide
|
153 |
-
4) Also try the bert models rather than directly using the mixtral model ()
|
154 |
-
5) Once the model is trained copy the checkpoint folder and paste in a local env
|
|
|
|
|
155 |
'''
|
|
|
1 |
+
|
2 |
+
print('this is the main file called main.py')
|
3 |
+
|
4 |
+
'''
|
5 |
+
|
6 |
#this repo contains the code for mixtral model for finding the icd-10 codes and this scripts runs well on the single GPU and is now trying to run with the multiple GPU and i need to make sure that this script runs in a multi gpu environment
|
7 |
|
8 |
import warnings
|
|
|
152 |
|
153 |
# final model prepared
|
154 |
'''
|
155 |
+
# 1) Make sure the model runs on multi gpu script !
|
156 |
+
# 2) The dataset is loaded
|
157 |
+
# 3) The langchain implementation to oversee the prompt generation guide
|
158 |
+
# 4) Also try the bert models rather than directly using the mixtral model ()
|
159 |
+
# 5) Once the model is trained copy the checkpoint folder and paste in a local env
|
160 |
+
'''
|
161 |
+
|
162 |
'''
|
maintain.py
CHANGED
@@ -1 +1,2 @@
|
|
1 |
-
print("This shit is working now ")
|
|
|
|
1 |
+
print("This shit is working now ")
|
2 |
+
print("Am i getting more errors now ? ")
|