Spaces:

ngxson
/

mergekit-extract-lora

Paused

ngxson HF Staff commited on Jan 9

Commit

af5ee17

verified ·

1 Parent(s): 2a26a04

Update Dockerfile

Files changed (1) hide show

Dockerfile CHANGED Viewed

@@ -22,7 +22,9 @@ RUN pip install --no-cache-dir -U pip setuptools wheel && \
                 bitsandbytes transformers
 COPY --chown=1000 . ${HOME}/app
-RUN git clone https://github.com/arcee-ai/mergekit.git && \
     cd mergekit && \
     pip install --upgrade pip && \
     pip install -e .

                 bitsandbytes transformers
 COPY --chown=1000 . ${HOME}/app
+# clone from ngxson's fork to support tied embeddings models, see https://github.com/arcee-ai/mergekit/pull/483
+RUN git clone https://github.com/ngxson/mergekit.git -b xsn/tied_embd && \
     cd mergekit && \
     pip install --upgrade pip && \
     pip install -e .