Unverified Commit 4cf51849 authored by narugo1992's avatar narugo1992 Committed by GitHub
Browse files

Merge pull request #76 from deepghs/dev/lora

dev(narugo): add model metadata reader and writer
parents cc802fc1 153de319
Loading
Loading
Loading
Loading
+1 −0
Original line number Diff line number Diff line
@@ -44,6 +44,7 @@ jobs:
          sudo apt-get install -y make wget curl cloc graphviz pandoc
          dot -V
          python -m pip install -r requirements.txt
          python -m pip install -r requirements-model.txt
          python -m pip install -r requirements-doc.txt
      - name: Prepare dataset
        uses: nick-fields/retry@v2
+1 −0
Original line number Diff line number Diff line
@@ -86,6 +86,7 @@ jobs:
          python -m pip install --upgrade pip
          pip install --upgrade flake8 setuptools wheel twine
          pip install -r requirements.txt
          pip install -r requirements-model.txt
          pip install -r requirements-test.txt
      - name: Test the basic environment
        shell: bash
+1 −0
Original line number Diff line number Diff line
@@ -10,4 +10,5 @@ imgutils.sd
    :maxdepth: 3

    metadata
    model
+22 −0
Original line number Diff line number Diff line
imgutils.sd.model
====================================

.. currentmodule:: imgutils.sd.model

.. automodule:: imgutils.sd.model


read_metadata
------------------------------------------

.. autofunction:: read_metadata



save_with_metadata
------------------------------------------

.. autofunction:: save_with_metadata


+1 −0
Original line number Diff line number Diff line
@@ -3,3 +3,4 @@ Overview:
    Utilities for dealing with data from `AUTOMATIC1111/stable-diffusion-webui <https://github.com/AUTOMATIC1111/stable-diffusion-webui>`_.
"""
from .metadata import parse_sdmeta_from_text, get_sdmeta_from_image, SDMetaData
from .model import read_metadata, save_with_metadata
Loading