{"id":47944268,"url":"https://github.com/dataiku/dss-plugin-ai-art","last_synced_at":"2026-04-04T08:20:53.565Z","repository":{"id":206242763,"uuid":"578521446","full_name":"dataiku/dss-plugin-ai-art","owner":"dataiku","description":"DSS plugin to generate images from text using Stable Diffusion","archived":false,"fork":false,"pushed_at":"2023-10-07T22:33:32.000Z","size":2001,"stargazers_count":1,"open_issues_count":0,"forks_count":0,"subscribers_count":19,"default_branch":"main","last_synced_at":"2023-11-08T16:45:30.245Z","etag":null,"topics":[],"latest_commit_sha":null,"homepage":"","language":"Python","has_issues":true,"has_wiki":null,"has_pages":null,"mirror_url":null,"source_name":null,"license":"apache-2.0","status":null,"scm":"git","pull_requests_enabled":true,"icon_url":"https://github.com/dataiku.png","metadata":{"files":{"readme":"README.md","changelog":"CHANGELOG.md","contributing":null,"funding":null,"license":"LICENSE","code_of_conduct":"CODE_OF_CONDUCT.md","threat_model":null,"audit":null,"citation":null,"codeowners":null,"security":null,"support":null,"governance":null}},"created_at":"2022-12-15T08:56:39.000Z","updated_at":"2023-11-08T16:45:35.599Z","dependencies_parsed_at":null,"dependency_job_id":"ab8dfe27-9b96-49ca-ba7f-b94e8fc1747f","html_url":"https://github.com/dataiku/dss-plugin-ai-art","commit_stats":null,"previous_names":["dataiku/dss-plugin-ai-art"],"tags_count":3,"template":null,"template_full_name":null,"purl":"pkg:github/dataiku/dss-plugin-ai-art","repository_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/dataiku%2Fdss-plugin-ai-art","tags_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/dataiku%2Fdss-plugin-ai-art/tags","releases_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/dataiku%2Fdss-plugin-ai-art/releases","manifests_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/dataiku%2Fdss-plugin-ai-art/manifests","owner_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners/dataiku","download_url":"https://codeload.github.com/dataiku/dss-plugin-ai-art/tar.gz/refs/heads/main","sbom_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/dataiku%2Fdss-plugin-ai-art/sbom","scorecard":null,"host":{"name":"GitHub","url":"https://github.com","kind":"github","repositories_count":286080680,"owners_count":31392747,"icon_url":"https://github.com/github.png","version":null,"created_at":"2022-05-30T11:31:42.601Z","updated_at":"2026-04-04T04:26:24.776Z","status":"ssl_error","status_checked_at":"2026-04-04T04:23:34.147Z","response_time":60,"last_error":"SSL_read: unexpected eof while reading","robots_txt_status":"success","robots_txt_updated_at":"2025-07-24T06:49:26.215Z","robots_txt_url":"https://github.com/robots.txt","online":false,"can_crawl_api":true,"host_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub","repositories_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories","repository_names_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repository_names","owners_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners"}},"keywords":[],"created_at":"2026-04-04T08:20:52.890Z","updated_at":"2026-04-04T08:20:53.550Z","avatar_url":"https://github.com/dataiku.png","language":"Python","readme":"# AI Art\nAI Art is a plugin for Dataiku that allows you to generate images from text prompts.\n\n## Usage\n\nWith this plugin, you will be able to:\n\n- Generate images from a prompt using **Stable Diffusion**\n- Generate images from another image and a prompt using **Stable Diffusion**\n\n## How to set up\n\n### CUDA\n\nA CUDA-capable GPU with at least 6 GB of VRAM is recommended. \nYou can run the plugin without a GPU, but it will be significantly slower.\n\nIf you’re using a CUDA GPU, your GPU must support CUDA 11.6, and the NVIDIA drivers must be installed on the server. \nFor installation instructions, see [NVIDIA Driver Installation Quickstart Guide](https://docs.nvidia.com/datacenter/tesla/tesla-installation-notes/index.html).\n\n### Download weights\n\nBefore you can use the plugin, you need to download pre-trained weights to a local managed folder. The plugin is designed to work with version 2 of the Stable Diffusion weights, although it may work with other versions and derivatives as well.\n\nFor example, if you run the below code in a notebook, it will download the Stability AI version 2.1 weights, and upload them to a new managed folder.\n\nThe code requires a code environment with the *huggingface-hub* package installed.\n\n##### Limitations and warnings\n\n- The weights available from Stability AI are licensed under the\nCreativeML OpenRAIL++-M license, which restricts usage. You can view the\nlicense [here](https://huggingface.co/stabilityai/stable-diffusion-2/raw/main/LICENSE-MODEL). Hence, the weights must be acquired manually before you can use this plugin. \n\n- The weights must be stored on the local filesystem. If a remote folder (S3,\n    etc) is used, or if the recipe uses containerized execution, the weights\n    will be downloaded to a temporary directory every time the recipe is run.\n    This is because the method used to load the weights ([from_pretrained])\n    requires a local filepath.\n\n[from_pretrained]: https://huggingface.co/docs/diffusers/v0.6.0/en/api/diffusion_pipeline#diffusers.DiffusionPipeline.from_pretrained\n\n\n\n```\nimport tempfile\nimport dataiku\nimport huggingface_hub\n\n# Name of the local folder that will be created\nFOLDER_NAME = \"my_weights\"\n# Hugging Face repository\nREPO = \"stabilityai/stable-diffusion-2-1\"\n# Revision of the repository\n#\n# If using the Stability AI weights, the \"fp16\" revision is recommended\n# if using CUDA. Otherwise, use the \"main\" revision.\nREVISION = \"fp16\"\n\nclient = dataiku.api_client()\nproject = client.get_default_project()\n\n# Create the managed folder\nfolder = project.create_managed_folder(FOLDER_NAME)\n\nwith tempfile.TemporaryDirectory(dir=\".\") as temp_dir:\n    # Download the weights to a temp local dir\n    huggingface_hub.snapshot_download(\n       repo_id=REPO,\n       revision=REVISION,\n       local_dir=temp_dir,\n       local_dir_use_symlinks=False,\n       # Skip large files that the plugin doesn't need\n       ignore_patterns=[\"*.ckpt\", \"*.safetensors\"],\n    )\n\n    # Upload the weights to the managed folder\n    folder.upload_folder(\"/\", temp_dir)\n```\n\n\nUsing a folder that’s stored on the local filesystem is recommended. If the folder is stored on a remote connection (Amazon S3, Google Cloud Storage, etc), the weights will be downloaded to a temporary directory every time the recipe is run.\n\n## How to use\n\nAI Art contains two methods for generating images: Text-to-Image\nGeneration, and Text-Guided Image-to-Image Generation\n\n### Text-to-Image Generation\n\nText-to-Image Generation is used to generate images from a text prompt.  \n\n1.  Create a *Text-to-Image Generation* recipe with your weights folderas the input.\n2.  Enter a text prompt in the *Prompt* field.\n3.  If you’re using the *fp16* revision of the weights, be sure to check the *Half precision* field.\n\n### Text-Guided Image-to-Image Generation\n\nText-Guided Image-to-Image Generation is used to modify an existing\nreference image based on a text prompt.  \n\n1.  Obtain a reference image that you want to use as a base, and upload it to a managed folder.\n2.  Create a *Text-Guided Image-to-Image Generation* recipe with your weights folder and your base-image folder as the inputs.\n3.  Enter a text prompt in the *Prompt* field.\n4.  Enter the path to your base image in the *Base image* field.\n5.  If you’re using the *fp16* revision of the weights, be sure to check the *Half precision* field.\n\n## Testing and linting\nFormat the code using Black:\n```bash\nmake black\n```\n\nLint the code using Flake8, and verify that the code was formatted using Black:\n```bash\nmake lint\n```\n\nRun unit tests:\n```bash\nmake unit-tests\n```\n\nRun integration tests:\n```bash\nmake integration-tests\n```\n\nRun all linters and tests at once:\n```bash\nmake tests\n```\n\n## Building\nCreate a plugin archive that can be imported into Dataiku:\n```bash\nmake plugin\n```\n\n## Known limitations\n1.  Due to licensing restrictions, the weights must be acquired manually before\n    you can use this plugin. See the [documentation][plugin_documentation] for\n    details.\n\n1.  The weights must be stored on the local filesystem. If a remote folder (S3,\n    etc) is used, or if the recipe uses containerized execution, the weights\n    will be downloaded to a temporary directory every time the recipe is run.\n    This is because the method used to load the weights ([from_pretrained])\n    requires a local filepath.\n\n[from_pretrained]: https://huggingface.co/docs/diffusers/v0.6.0/en/api/diffusion_pipeline#diffusers.DiffusionPipeline.from_pretrained","funding_links":[],"categories":[],"sub_categories":[],"project_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fdataiku%2Fdss-plugin-ai-art","html_url":"https://awesome.ecosyste.ms/projects/github.com%2Fdataiku%2Fdss-plugin-ai-art","lists_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fdataiku%2Fdss-plugin-ai-art/lists"}