ProCreations commited on
Commit
fc2119a
·
verified ·
1 Parent(s): d827cd9

Pin plain NVFP4 build, source, and bounded validation plan

Browse files
Files changed (1) hide show
  1. reproducibility/build.py +2 -2
reproducibility/build.py CHANGED
@@ -56,11 +56,11 @@ def main():
56
  REPORT['hardware'] = [torch.cuda.get_device_name(0)]
57
  record()
58
  from transformers.dynamic_module_utils import get_class_from_dynamic_module
59
- source = Path(snapshot_download(SOURCE, revision=REV, allow_patterns=['*.py','config.json'], max_workers=8))
60
  cls = get_class_from_dynamic_module('modeling_agnes.AgnesForConditionalGeneration',source)
61
  assert cls.__name__ == 'AgnesForConditionalGeneration'
62
  print('AGNES_REMOTE_IMPLEMENTATION_IMPORT_PASS',flush=True)
63
- source = Path(snapshot_download(SOURCE, revision=REV, max_workers=8))
64
  src_index = json.loads((source/'model.safetensors.index.json').read_text())
65
  src_map = src_index['weight_map']
66
  model = AutoModelForImageTextToText.from_pretrained(source, trust_remote_code=True,
 
56
  REPORT['hardware'] = [torch.cuda.get_device_name(0)]
57
  record()
58
  from transformers.dynamic_module_utils import get_class_from_dynamic_module
59
+ source = Path(snapshot_download(SOURCE, revision=REV, local_dir=ROOT/'source', allow_patterns=['*.py','config.json'], max_workers=8))
60
  cls = get_class_from_dynamic_module('modeling_agnes.AgnesForConditionalGeneration',source)
61
  assert cls.__name__ == 'AgnesForConditionalGeneration'
62
  print('AGNES_REMOTE_IMPLEMENTATION_IMPORT_PASS',flush=True)
63
+ source = Path(snapshot_download(SOURCE, revision=REV, local_dir=ROOT/'source', max_workers=8))
64
  src_index = json.loads((source/'model.safetensors.index.json').read_text())
65
  src_map = src_index['weight_map']
66
  model = AutoModelForImageTextToText.from_pretrained(source, trust_remote_code=True,