inoki-giskard commited on
Commit
fe45df8
·
1 Parent(s): 46fe2b3

Add requirements for demo Space external worker

Browse files
Files changed (3) hide show
  1. Dockerfile +3 -0
  2. hf.sh +1 -1
  3. requirements.txt +3 -0
Dockerfile CHANGED
@@ -28,6 +28,9 @@ USER giskard
28
  # Only changed the original supervisord.conf to redirect logs to stdout
29
  COPY --chown=giskard supervisord.conf $GSK_DIST_PATH
30
 
 
 
 
31
  # Internal workr doesn't download resource, but takes them directly from projects, so no cache
32
  ENV GSK_CACHE_DIR=projects
33
 
 
28
  # Only changed the original supervisord.conf to redirect logs to stdout
29
  COPY --chown=giskard supervisord.conf $GSK_DIST_PATH
30
 
31
+ # Copy requirements for demo Space external worker
32
+ COPY --chown=giskard requirements.txt $GSK_DIST_PATH
33
+
34
  # Internal workr doesn't download resource, but takes them directly from projects, so no cache
35
  ENV GSK_CACHE_DIR=projects
36
 
hf.sh CHANGED
@@ -33,7 +33,7 @@ autorestart=true
33
  redirect_stderr=true
34
  startsecs=0
35
  startretries=5000
36
- command=/bin/bash -c '[ ! -z \"\$GISKARD_DEFAULT_API_KEY\" ] && python -m giskard.cli worker start -u \"http://localhost:9000\" -k \"\$GISKARD_DEFAULT_API_KEY\"; sleep 5'
37
  """ >> "${GSK_DIST_PATH}/supervisord.conf"
38
  fi
39
 
 
33
  redirect_stderr=true
34
  startsecs=0
35
  startretries=5000
36
+ command=/bin/bash -c 'pip install -r ${GSK_DIST_PATH}/requirements.txt && python -m giskard.cli worker start -u \"http://localhost:9000\" -k \"\$GISKARD_DEFAULT_API_KEY\"; sleep 5'
37
  """ >> "${GSK_DIST_PATH}/supervisord.conf"
38
  fi
39
 
requirements.txt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ transformers==4.33.2
2
+ datasets==2.14.5
3
+ torch==2.0.1