Skip to content

Conversation

@ishan-marikar
Copy link

  • Created .dockerignore to exclude unnecessary files from Docker context.
  • Added Dockerfile.app for the Gradio web app with Python 3.10 and necessary dependencies.
  • Added Dockerfile.cli for the command-line interface with similar setup.
  • Added docker-compose.yml to manage app and CLI services with Nvidia GPU support.

- Created .dockerignore to exclude unnecessary files from Docker context.
- Added Dockerfile.app for the Gradio web app with Python 3.10 and necessary dependencies.
- Added Dockerfile.cli for the command-line interface with similar setup.
- Added docker-compose.yml to manage app and CLI services with GPU support.
@ishan-marikar
Copy link
Author

Running the Gradio Web App

Build and start the web app using the "app" profile:

docker-compose --profile app up

Access the Gradio interface at http://localhost:7860

Using the CLI Tool

Build the CLI container using the "cli" profile:

docker-compose --profile cli build

Run it with arguments (by default --help is passed), specifying the output path in the mounted /app/data directory:

docker-compose --profile cli run dia-cli "[S1] Hello, welcome to Dia! [S2] Thanks for trying out our text to speech model." --output /app/data/output.wav

After running this command, the generated audio file will be available at ./data/output.wav on your host system.

For one-off commands, you can also use:

docker run --gpus all -v $(pwd)/data:/app/data dia-cli "[S1] Hello! [S2] Welcome!" --output /app/data/output.wav

@ishan-marikar
Copy link
Author

I'm currently adding support for cpu and mps as well.

@jaehong21
Copy link
Collaborator

We are now going through #46 for the newly added Dockerfile.
So it might be needed to go through its content. currently it has Dockerfile.gpu and Dockerfile.cpu

But, still seems to lack ARM and mps support like you mentioned.
Currently planning #46 to be merged in days. and it will be nice to continue your contribution with updated main branch!

+) love the idea seperating the CLI and gradio part with dockerfile
+) and the docker-compose.yaml

will come back to you after #46 this one.

@ishan-marikar
Copy link
Author

Of course, I'd be happy to contribute to #46. I'll move over some of the improvements to the mentioned PR along with ARM/MPS support (I'm in a position to test it as well).

@jaehong21
Copy link
Collaborator

@ishan-marikar
#46 is now merged! you can continue your work with existing dockerfiles in docker/ directory in project root!

+) and suggest to run ruff format before merge

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.

2 participants