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.

Sign up for GitHub

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

Jump to bottom

Ben/sd1.4 #163

Closed
wants to merge 15 commits into from
Closed

Ben/sd1.4 #163

wants to merge 15 commits into from

Conversation

bgoelTT
Copy link
Contributor

@bgoelTT bgoelTT commented Jan 27, 2025 •
edited
Loading

This PR integrates the stable-diffusion-1.4 inference server into TT-Studio. The features include:

  • launching & destroying the server from the studio UI
  • a new Image Generation page
    • agnostic to the model being used (allows for future model integration such as a better stable-diffusion version, i.e., sd-3.5)
    • looks and functions similar to the ChatUI where the user submits prompts to the server and the UI "replies" back to the user with their generated image

Some future improvements:

  • adding a stateful chat system similar to ChatUI (currently if you navigate to a different page all the state information is destroyed)
  • Swapping the current initial sample image UI with something similar to ChatUI (example sample prompts to show the user how to generate the best images)
  • enable queuing of multiple image generation requests
    • the server can already handle this, needs frontend support

bgoelTT and others added 6 commits January 23, 2025 12:50
@bgoelTT
Add stable-diffusion-1.4 to model_config.py
180731e
@anirudTT @tstescoTT @bgoelTT
Anirud/front end UI changes (#21)
bc589ad 
* change font and app name

* remove dropdown from theme toggle button

* add shadow effects on hover

* fix :@import must precede all other statements

* create and use common theme class for navbar

* add help sidebar component :
- opens and closes on clicking help icon on navbar
- inital jsx for component

* break down model selection component into multiple components

* add more common themes

* update help component

* Add more on hover effects for buttons
- on hover change button color from grey to white

* add TT font

* add model name

* more fixes

* more fixes to chatui
- break long texts
- some theme providers

* add menu bar shadcnn component

* shadcnn menu bar component

* send model name to chatui component

* make navbar vertical in chat ui mode

* add breadcumbs links on chat ui page

* ensure mode name is being sent to chat page for display

* Ui color fixes for main page footer

* make default theme as dark mode

* clean up

* fix help sidebar to appear on whole page

* update package lock json to fix:
- ETIMEDOUT errors

---------

Co-authored-by: Tom Stesco <tstesco@tenstorrent.com>
@bgoelTT
Fix image generation tooltip in navbar
22087e7
@bgoelTT
Add ImageGeneration modeltype to navbar and fix ModelID propagation
34087d4
@bgoelTT
Fully hook up frontend with backend
103c939
@bgoelTT
Propagate 503 errors to frontend
f81702e
@bgoelTT bgoelTT changed the base branch from main to dev January 28, 2025 15:53
@bgoelTT bgoelTT marked this pull request as ready for review February 2, 2025 21:16
@bgoelTT bgoelTT requested a review from anirudTT February 2, 2025 21:16
@bgoelTT bgoelTT self-assigned this Feb 2, 2025
@bgoelTT
Copy link
Contributor Author

bgoelTT commented Feb 24, 2025

Superceded by #193

@bgoelTT bgoelTT closed this Feb 24, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@anirudTT anirudTT Awaiting requested review from anirudTT

Assignees

@bgoelTT bgoelTT

Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
2 participants
@bgoelTT @anirudTT