Spaces:
Sleeping
Sleeping
File size: 2,966 Bytes
1282ba1 a62d768 1282ba1 a62d768 1282ba1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 |
{
"cells": [
{
"cell_type": "markdown",
"metadata": {},
"source": [
"# Compile StyleForge CUDA Kernels\n",
"\n",
"Run this notebook in Google Colab to compile the CUDA kernels and download them for deployment to Hugging Face Spaces."
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"# Check if GPU is available\n",
"!nvidia-smi"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"# Clone the repository\n",
"!git clone https://github.com/olivialiau/StyleForge\n",
"%cd StyleForge/huggingface-space"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"# The PyTorch with CUDA should already be installed in Colab\n",
"import torch\n",
"print(f'PyTorch: {torch.__version__}')\n",
"print(f'CUDA available: {torch.cuda.is_available()}')\n",
"if torch.cuda.is_available():\n",
" print(f'CUDA version: {torch.version.cuda}')\n",
" print(f'GPU: {torch.cuda.get_device_name(0)}')"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"# Install ninja for compilation\n",
"!pip install ninja -q"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"# Run the compilation script\n",
"!python compile_kernels.py"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"# List the compiled kernels\n",
"!ls -lh kernels/prebuilt/"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"# Download the compiled kernels\n",
"from google.colab import files\n",
"import pathlib\n",
"\n",
"prebuilt_dir = pathlib.Path('kernels/prebuilt')\n",
"kernel_files = list(prebuilt_dir.glob('*.so')) + list(prebuilt_dir.glob('*.pyd'))\n",
"\n",
"if kernel_files:\n",
" for f in kernel_files:\n",
" print(f'Downloading: {f.name}')\n",
" files.download(str(f))\n",
"else:\n",
" print('No kernel files found!')\n",
" print('Check the compilation output above for errors.')"
]
}
],
"metadata": {
"kernelspec": {
"display_name": "Python 3",
"language": "python",
"name": "python3"
},
"language_info": {
"name": "python",
"version": "3.10.0"
}
},
"nbformat": 4,
"nbformat_minor": 4
}
|