File size: 386 Bytes
25bcc11
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
#!/bin/bash
# Additional setup for chat_env
set -e

# Install Python dependencies
pip install --no-cache-dir -r /tmp/requirements.txt

# Set up cache directory for Hugging Face models
mkdir -p /.cache && chmod 777 /.cache

# Pre-download the GPT-2 model to avoid permission issues during runtime
python -c "from transformers import GPT2Tokenizer; GPT2Tokenizer.from_pretrained('gpt2')"