more training updates

This commit is contained in:
Alex O'Connell
2025-11-30 15:31:58 -05:00
parent 1a5445e68a
commit 9f51dd0e94
4 changed files with 93 additions and 25 deletions

View File

@@ -8,6 +8,11 @@ from typing import Callable
from tqdm import tqdm
import webcolors
# ensure we can import from the data/ directory
import os
import sys
sys.path.append(os.path.dirname(os.path.abspath(__file__)))
from device_types import *
from prompting import generate_system_prompt, USER_INSTRUCTION_PROMPT
from utils import get_random_response, generate_random_parameter, closest_color, get_dataset_piles, NoResponseAvailableException