tanner/min-dalle-test
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

mega model in standard runtime

Browse Source
This commit is contained in:
Brett Kuprel 2022-06-30 09:43:13 -04:00
parent e9c01e32a5
commit 6d1ac07442
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
min_dalle.ipynb vendored
View File

@ -70,7 +70,8 @@
"torch = True #@param {type:\"boolean\"}\n", "torch = True #@param {type:\"boolean\"}\n",
"reusable = False #@param {type:\"boolean\"}\n", "reusable = False #@param {type:\"boolean\"}\n",
"model_class = MinDalleTorch if torch else MinDalleFlax\n", "model_class = MinDalleTorch if torch else MinDalleFlax\n",
"model = model_class(mega, not expendable)\n" "is_expendable = not reusable\n",
"model = model_class(mega, is_expendable)\n"
], ],
"metadata": { "metadata": {
"id": "8W-L2ICFGFup", "id": "8W-L2ICFGFup",
@ -79,7 +80,7 @@
"base_uri": "https://localhost:8080/" "base_uri": "https://localhost:8080/"
} }
}, },
"execution_count": 6, "execution_count": null,
"outputs": [ "outputs": [
{ {
"output_type": "stream", "output_type": "stream",
@ -105,7 +106,7 @@
}, },
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": 7, "execution_count": null,
"metadata": { "metadata": {
"colab": { "colab": {
"base_uri": "https://localhost:8080/", "base_uri": "https://localhost:8080/",
@ -172,7 +173,7 @@
], ],
"name": "min-dalle", "name": "min-dalle",
"provenance": [], "provenance": [],
"authorship_tag": "ABX9TyMtWeoRGqaMmLjAXcNCf4AW", "authorship_tag": "ABX9TyPWA14jCgKCRzp2RXXzxZ+q",
"include_colab_link": true "include_colab_link": true
}, },
"gpuClass": "standard", "gpuClass": "standard",