|
@@ -1,58 +0,0 @@
|
|
|
-{
|
|
|
- "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
|
|
|
-}
|