Skip to content
Closed
Show file tree
Hide file tree
Changes from 2 commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions apps/pii/requirements-gliner.txt
Original file line number Diff line number Diff line change
Expand Up @@ -6,5 +6,5 @@
# torch is pinned in the Dockerfile instead: the CPU and CUDA targets install
# the same version from different wheel indexes.
gliner==0.2.27
transformers==5.3.0
huggingface_hub==1.3.0
transformers==5.5.0
Comment thread
waleedlatif1 marked this conversation as resolved.
huggingface_hub==1.23.0
9 changes: 8 additions & 1 deletion docker/pii.Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -87,9 +87,16 @@ RUN --mount=type=cache,target=/root/.cache/pip \
# Bake the GLiNER weights at build time (cached layer) so startup never
# touches the network. HF_HUB_OFFLINE makes a missing/overridden
# PII_GLINER_MODEL fail fast at startup instead of silently downloading.
# The predict_entities smoke assert (labels from engines.py's
# GLINER_ENTITY_MAPPING) fails the build if a transformers/gliner bump loads
# the model fine but silently stops returning detections.
ENV HF_HOME=/opt/hf-cache
ARG GLINER_MODEL=urchade/gliner_multi_pii-v1
RUN python -c "from gliner import GLiNER; GLiNER.from_pretrained('${GLINER_MODEL}')" && \
RUN python -c "from gliner import GLiNER; \
model = GLiNER.from_pretrained('${GLINER_MODEL}'); \
ents = model.predict_entities('John Smith flew to Paris on 4 May 2021.', ['person', 'location', 'date']); \
labels = {e['label'] for e in ents}; \
assert {'person', 'location'} <= labels, f'GLiNER smoke inference failed, got: {ents}'" && \
Comment thread
waleedlatif1 marked this conversation as resolved.
Outdated
chmod -R a+rX /opt/hf-cache
ENV HF_HUB_OFFLINE=1

Expand Down
Loading