Pin plain NVFP4 build, source, and bounded validation plan
Browse files- 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,
|