Databricks Community

gary7135 · 07-20-2023

I am trying out the new Meta LLama2 model.Following the databricks provided notebook example: https://github.com/databricks/databricks-ml-examples/blob/master/llm-models/llamav2/llamav2-13b/01_load_inference.py I keep getting CUDA out of memory. My G...

gary7135 · 04-11-2023

When using GridsearchCV from spark-sklearn, I got GridSearchCV giving " __init__() got an unexpected keyword argument 'fit_params' errorI am using sklearn 1.2.2 and spark-sklearn 0.3.0I think this is because spark-sklearn GridsearchCV still has the f...

gary7135 · 07-21-2023

Thank you. I am running this python file directly in Databricks notebook https://github.com/databricks/databricks-ml-examples/blob/master/llm-models/llamav2/llamav2-7b/01_load_inference.pyThe file does not seem to reference the config json file?

gary7135 · 07-21-2023

Thank you. Can you provide example of how to set this argument in notebooks?

Databricks Community

User Stats

User Activity

CUDA out of memory

Unable to use GridsearchCV from spark-sklearn due to 'fit_params' error

Re: CUDA out of memory

Re: CUDA out of memory