Skip to content

Commit

Permalink
Fix civitai URL prefix
Browse files Browse the repository at this point in the history
  • Loading branch information
yoland68 committed May 15, 2024
1 parent 6ed7814 commit 0fdb3ac
Show file tree
Hide file tree
Showing 3 changed files with 27 additions and 2 deletions.
25 changes: 25 additions & 0 deletions .github/workflows/pytest.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,25 @@
name: Python application

on: [push, pull_request]

jobs:
build:
runs-on: ubuntu-latest

steps:
- uses: actions/checkout@v2

- name: Set up Python
uses: actions/setup-python@v2
with:
python-version: '3.8' # Adjust the version as needed

- name: Install dependencies
run: |
python -m pip install --upgrade pip
pip install pytest
pip install -r requirements.txt # If you have other dependencies
- name: Run tests
run: |
pytest
2 changes: 1 addition & 1 deletion .pre-commit-config.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -9,4 +9,4 @@ repos:
hooks:
- id: pylint
args:
- --disable=R,C,W,E0401
- --disable=R,C,W,E0401
2 changes: 1 addition & 1 deletion comfy_cli/command/models/models.py
Original file line number Diff line number Diff line change
Expand Up @@ -39,7 +39,7 @@ def check_civitai_url(url: str) -> Tuple[bool, bool, int, int]:
model_id: The model id or None if it's api url
version_id: The version id or None if it doesn't have version id info
"""
prefix = "civitai.co"
prefix = "civitai.com"
try:
if prefix in url:
# URL is civitai api download url: https://civitai.com/api/download/models/12345
Expand Down

0 comments on commit 0fdb3ac

Please sign in to comment.