diff --git a/demo-notebooks/additional-interactive/hf_interactive.ipynb b/demo-notebooks/additional-interactive/hf_interactive.ipynb index e80df90db..e5282cd39 100644 --- a/demo-notebooks/additional-interactive/hf_interactive.ipynb +++ b/demo-notebooks/additional-interactive/hf_interactive.ipynb @@ -53,11 +53,13 @@ "metadata": {}, "outputs": [], "source": [ - "# Create authentication object for oc user permissions and login\n", + "# Create authentication object for user permissions\n", + "# IF unused, SDK will automatically check for default kubeconfig, then in-cluster config\n", + "# KubeConfigFileAuthentication can also be used to specify kubeconfig path manually\n", "auth = TokenAuthentication(\n", " token = \"XXXX\",\n", " server = \"XXXX\",\n", - " skip_tls = True\n", + " skip_tls = False\n", ")\n", "auth.login()" ] @@ -88,13 +90,12 @@ "# Create our cluster and submit appwrapper\n", "cluster = Cluster(ClusterConfiguration(name='hfgputest', \n", " namespace=\"default\",\n", - " min_worker=1, \n", - " max_worker=1, \n", + " num_workers=1,\n", " min_cpus=8, \n", " max_cpus=8, \n", " min_memory=16, \n", " max_memory=16, \n", - " gpu=4,\n", + " num_gpus=4,\n", " image=\"quay.io/project-codeflare/ray:2.5.0-py38-cu116\",\n", " instascale=True, machine_types=[\"m5.xlarge\", \"p3.8xlarge\"]))" ] @@ -319,7 +320,7 @@ "#install additional libraries that will be required for this training\n", "runtime_env = {\"pip\": [\"transformers\", \"datasets\", \"evaluate\", \"pyarrow<7.0.0\", \"accelerate\"]}\n", "\n", - "ray.init(address=f'{ray_cluster_uri}', runtime_env=runtime_env)\n", + "ray.init(address=ray_cluster_uri, runtime_env=runtime_env)\n", "\n", "print(\"Ray cluster is up and running: \", ray.is_initialized())" ]