Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

AI-Image-Generation\Offline\Stable-Deffusion-Model 🚀 #242

Open
wants to merge 1 commit into
base: main
Choose a base branch
from

Conversation

ParagGhatage
Copy link

@ParagGhatage ParagGhatage commented Jan 16, 2025

Fixes: #232

Me and @Karn-x7 worked on this together and made this possible with couple of sleepless night.

PR OVERVIEW
This PR basically adds custom AI image generation which would work offline and would generate images from the user prompt .

Our Approach
We used one of the stable diffusion models which we trained, tweaked, fine tuned - a lot of optimization .
we took the model which was consuming the space around 8GB storage,4GB ram(for working)
And now guess what -
-This model only takes around 4GB STORAGE and ram less than 500mb.
Screenshot 2025-01-17 023749

-And wrapped the backend implementation in Fast API for seamless integration.

{Frontend}
-And for the frontend we created a very sweet, subtle, landing page which is highly interactive
With some great animation like you can see a star bouncing and many more .

Screenshot_2025-01-17_033445

-Which is dynamically adaptive to Dark Mode.
-And added a dynamic promt functionality to enhance UX.

Working Video-
BTW, this prompt was his idea. @Karn-x7 😂

image-generation.mp4

Sample images-

Prompt: Astronaut in a jungle, cold color palette, muted colors, detailed, 8k

generated_image (14)

Prompt: School in cosmic style

Screenshot_2025-01-17_041505

Prompt: Desert view in realistic style

Screenshot_2025-01-17_041823

@Karn-x7
Copy link
Contributor

Karn-x7 commented Jan 16, 2025

bro you forgot some secret ingredients.🤯
-doses of coffee intake
-bro my waking up alarm has turned to be my sleeping alarm.
-cooked laptop .

Screenshot 2025-01-17 043320
Screenshot_2025-01-17_043346

Working Flowchart @Pranav0-0Aggarwal
Screenshot 2025-01-16 180304

@ParagGhatage
Copy link
Author

But it was worth every second spent.

@ParagGhatage
Copy link
Author

#243 adds setup instructions.

@ParagGhatage
Copy link
Author

I have added PR with detailed documentation of an image-generation model. #244

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Feat: Ai image generation/locally-offline
2 participants