Support for AutoModelForCausalLM

GoogleCloudPlatform / llm-pipeline-examples

Apache License 2.0

114 stars 27 forks source link

Open sagar-deepscribe opened 1 year ago

sagar-deepscribe commented 1 year ago

Can there be an image that can support CausalLM training ? This is important for our use-case and I was wondering if you could suggest some steps to:

Create my own container image with modified train and dataset read code (something similar to what you did to create this pipeline)