(*args, **kwargs)
| 118 | |
| 119 | @wraps(func) |
| 120 | def wrapper(*args, **kwargs): |
| 121 | with ( |
| 122 | patch.dict(os.environ, {"HF_HUB_OFFLINE": "1"}), |
| 123 | patch( |
| 124 | "tensorrt_llm.llmapi.utils.snapshot_download", side_effect=mock_snapshot_download |
| 125 | ), |
| 126 | ): |
| 127 | return func(*args, **kwargs) |
| 128 | |
| 129 | return wrapper |