diff --git a/torchbenchmark/models/torch_multimodal_clip/install.py b/torchbenchmark/models/torch_multimodal_clip/install.py index 0723509634..91c9f9dbad 100644 --- a/torchbenchmark/models/torch_multimodal_clip/install.py +++ b/torchbenchmark/models/torch_multimodal_clip/install.py @@ -5,7 +5,10 @@ def download(output_filename, uri): # Download the file with streaming to handle large files - response = requests.get(uri, stream=True) + headers = { + 'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.3' + } + response = requests.get(uri, headers=headers, stream=True) if response.status_code == 200: with open(output_filename, 'wb') as f: for chunk in response.iter_content(chunk_size=8192): # Define the chunk size to be used