{ "cells": [ { "cell_type": "code", "execution_count": null, "metadata": {}, "outputs": [ { "ename": "", "evalue": "", "output_type": "error", "traceback": [ "\u001b[1;31mRunning cells with 'Python 3.8.0 ('petals')' requires ipykernel package.\n", "\u001b[1;31mRun the following command to install 'ipykernel' into the Python environment. \n", "\u001b[1;31mCommand: 'conda install -n petals ipykernel --update-deps --force-reinstall'" ] } ], "source": [ "import os\n", "import sys\n", "sys.path.insert(0, \"..\")\n", " \n", "import torch\n", "import transformers\n", "\n", "# Import a Petals model\n", "from src.client.remote_model import DistributedBloomForCausalLM" ] }, { "cell_type": "code", "execution_count": null, "metadata": {}, "outputs": [], "source": [] } ], "metadata": { "kernelspec": { "display_name": "Python 3.8.0 ('petals')", "language": "python", "name": "python3" }, "language_info": { "name": "python", "version": "3.8.0" }, "orig_nbformat": 4, "vscode": { "interpreter": { "hash": "a303c9f329a09f921588ea6ef03898c90b4a8e255a47e0bd6e36f6331488f609" } } }, "nbformat": 4, "nbformat_minor": 2 }