File size: 231 Bytes
3f7cfab
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
#!/bin/bash

echo "$(date '+%F %T') BEGIN: run-gpt.sh"

set -e

export TRANSFORMERS_CACHE=/h2ogpt_env/.cache

# run generate.py
mkdir -p /h2ogpt_env && cd /h2ogpt_env
exec python3.10 /workspace/generate.py --base_model=${HF_MODEL}