Skip to content

Commit 0181b4b

Browse files
committed
local upload server can be called cluster.local in addition to .internal
Signed-off-by: technillogue <technillogue@gmail.com>
1 parent c5ad13b commit 0181b4b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

python/cog/server/clients.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -186,7 +186,7 @@ async def chunk_file_reader() -> AsyncIterator[bytes]:
186186
# and is critical for upload training/quantization outputs
187187
# if we get multipart uploads working or a separate API route
188188
# then we could drop this
189-
if url and ".internal" in url:
189+
if url and (".internal" in url or ".local" in url):
190190
log.info("doing test upload to %s", url)
191191
resp1 = await self.file_client.put(
192192
url,

0 commit comments

Comments
 (0)