Added WD14 Auto Caption task#15
Open
ParamThakkar123 wants to merge 6 commits into
Open
Conversation
deep1401
requested changes
Feb 2, 2026
deep1401
requested changes
Feb 4, 2026
Member
deep1401
left a comment
There was a problem hiding this comment.
You will need to convert the task.json to task.yaml now, please refer here: https://lab.cloud/for-teams/task-yaml/task-structure
deep1401
requested changes
Feb 5, 2026
| name: auto-caption-wd14 | ||
| resources: | ||
| accelerators: "RTX3090:1" | ||
| setup: "uv pip install transformerlab torchvision huggingface-hub" |
Member
There was a problem hiding this comment.
You're importing datasets but dont install it. Maybe also look at installing the correct version of transformers
…ab-examples into add/auto-caption
deep1401
requested changes
Apr 17, 2026
| import tempfile | ||
| from datetime import datetime | ||
| from PIL import Image, ImageDraw, ImageFont, ImageFilter | ||
| import tensorflow as tf |
Member
There was a problem hiding this comment.
Do we need tensorflow somewhere?
If not then maybe we shouldn't import it?
I started a task and it failed because tensorflow was not installed and then when I added tensorflow to the uv pip install setup stuff and ran another job, I got the following error:
WARNING: All log messages before absl::InitializeLog() is called are written to STDERR
I0000 00:00:1776442839.004594 9195 cpu_feature_guard.cc:227] This TensorFlow binary is optimized to use available CPU instructions in performance-critical operations.
To enable the following instructions: AVX2 FMA, in other operations, rebuild TensorFlow with the appropriate compiler flags.
Traceback (most recent call last):
File "/home/sky/auto-caption-wd14/main.py", line 23, in <module>
from huggingface_hub import from_pretrained_keras, hf_hub_download
ImportError: cannot import name 'from_pretrained_keras' from 'huggingface_hub' (/home/sky/miniconda3/lib/python3.10/site-packages/huggingface_hub/__init__.py)
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
No description provided.