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 ? ")
|