complete-dope commited on
Commit
17ae86e
1 Parent(s): 1fd6029

dockerSecond

Browse files
Files changed (2) hide show
  1. main.py +12 -5
  2. 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 ? ")