https://github.com/lucataco/cog-ssd-1b

Attempt at cog wrapper for segmind/SSD-1B
https://github.com/lucataco/cog-ssd-1b

cog

Last synced: 12 months ago
JSON representation

Attempt at cog wrapper for segmind/SSD-1B

• Host: GitHub
• URL: https://github.com/lucataco/cog-ssd-1b
• Owner: lucataco
• Created: 2023-10-25T01:10:28.000Z (over 2 years ago)
• Default Branch: main
• Last Pushed: 2023-12-11T21:18:52.000Z (over 2 years ago)
• Last Synced: 2024-05-01T19:56:03.588Z (about 2 years ago)
• Topics: cog
• Language: Python
• Homepage: https://replicate.com/lucataco/ssd-1b
• Size: 5.53 MB
• Stars: 10
• Watchers: 1
• Forks: 1
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

          
# segmind/SSD-1BCog model

This is an implementation of the [segmind/SSD-1B](https://huggingface.co/segmind/SSD-1B) as a Cog model. [Cog packages machine learning models as standard containers.](https://github.com/replicate/cog)

First, download the pre-trained weights:

    cog run script/download-weights

Then, you can run predictions:

    cog predict -i prompt="with smoke, half ice and half fire and ultra realistic in detail.wolf, typography, dark fantasy, wildlife photography, vibrant, cinematic and on a black background" -i seed=36446545872

Or img2img:

    cog predict -i image=@output.0.png -i prompt="a wolf with pink and blue fur" -i seed=21272 -i disable_safety_checker=True

## Examples:

txt2img

![alt text](output.0.png)

img2img

![alt text](output.img2img.png)