-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathrun1.sh
More file actions
29 lines (22 loc) · 1.17 KB
/
run1.sh
File metadata and controls
29 lines (22 loc) · 1.17 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
#!/bin/bash
#BEFORE RUN IT, RUN ON TERMINAL ----> chmod +x run.sh
CONFIG_FILE="./config.env"
# HF_TOKEN=$(head -n 1 tokens.txt) # a priemira linha é o token do HF
TOKEN="$HF_TOKEN"
if [ -z "$TOKEN" ]; then
echo "Erro: variável de ambiente HF_TOKEN não está definida."
exit 1
fi
INPUT_PATH="logs/"
MODEL_ID="mistralai/Ministral-8B-Instruct-2410"
OUTPUT_PATH="dataset_newsmet/mistral/"
PROMPT_ID="prompt3"
CUDA_VISIBLE_DEVICES=1 python3 generate/genENtoPT_locais.py --model_id $MODEL_ID --hf_token $HF_TOKEN --output_path $OUTPUT_PATH --prompt_id $PROMPT_ID > "dataset_newsmet/mistral/mistral-log.txt" 2>&1
MODEL_ID="Qwen/Qwen2.5-7B-Instruct"
OUTPUT_PATH="dataset_newsmet/qwen/"
PROMPT_ID="prompt4"
CUDA_VISIBLE_DEVICES=1 python3 generate/genENtoPT_locais.py --model_id $MODEL_ID --hf_token $HF_TOKEN --output_path $OUTPUT_PATH --prompt_id $PROMPT_ID > "dataset_newsmet/qwen/qwen-log.txt" 2>&1
MODEL_ID="google/gemma-3-12b-it"
OUTPUT_PATH="dataset_newsmet/gemma3/"
PROMPT_ID="prompt3"
CUDA_VISIBLE_DEVICES=0 python3 generate/genENtoPT_locais.py --model_id $MODEL_ID --hf_token $HF_TOKEN --output_path $OUTPUT_PATH --prompt_id $PROMPT_ID > "dataset_newsmet/gemma3/gemma3-log.txt" 2>&1