Skip to content

Instantly share code, notes, and snippets.

@cvpfus
Created March 22, 2024 23:28
Show Gist options
  • Save cvpfus/a75169333d6b20f219750fe70007a994 to your computer and use it in GitHub Desktop.
Save cvpfus/a75169333d6b20f219750fe70007a994 to your computer and use it in GitHub Desktop.
#!/usr/bin/env bash
PRETRAINED=${1:-"hpcaitech/grok-1"}
TOKENIZER=${2:-"tokenizer.model"}
while true; do
echo "Enter the text you want to process:"
read USER_TEXT
python3 inference.py --pretrained "$PRETRAINED" \
--tokenizer "$TOKENIZER" \
--max_new_tokens 64 \
--text "$USER_TEXT"
done
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment