New transformers for qwen3 vl stuff

This commit is contained in:
Jake Poznanski 2025-10-24 16:56:39 +00:00
parent ba9e14154e
commit f31f68a4d2

View File

@ -102,7 +102,7 @@ b = Beaker.from_env(default_workspace="ai2/olmocr")
dataset_name = dataset.rstrip('/').split('/')[-1] dataset_name = dataset.rstrip('/').split('/')[-1]
commands = [ commands = [
"pip install .[train]", "pip install .[train]",
"pip install transformers==4.55.2", "pip install transformers==4.57.1",
"pip install flash-attn==2.8.0.post2 --no-build-isolation", "pip install flash-attn==2.8.0.post2 --no-build-isolation",
"pip install s5cmd", "pip install s5cmd",
f"s5cmd sync {dataset}/processed_* /data/{dataset_name}/", f"s5cmd sync {dataset}/processed_* /data/{dataset_name}/",