feat(colab): change defaults
Browse files
tools/inference/inference_pipeline.ipynb
CHANGED
|
@@ -141,9 +141,7 @@
|
|
| 141 |
{
|
| 142 |
"cell_type": "code",
|
| 143 |
"source": [
|
| 144 |
-
"params = model.to_fp16(params)
|
| 145 |
-
"#vqgan._params = vqgan.to_bf16(vqgan._params)\n",
|
| 146 |
-
"#clip._params = clip.to_bf16(clip._params)"
|
| 147 |
],
|
| 148 |
"metadata": {
|
| 149 |
"id": "2r-zWmsPqTjh"
|
|
@@ -354,7 +352,7 @@
|
|
| 354 |
"# We can customize top_k/top_p used for generating samples\n",
|
| 355 |
"gen_top_k = None\n",
|
| 356 |
"gen_top_p = None\n",
|
| 357 |
-
"temperature =
|
| 358 |
"cond_scale = 3.0"
|
| 359 |
]
|
| 360 |
},
|
|
|
|
| 141 |
{
|
| 142 |
"cell_type": "code",
|
| 143 |
"source": [
|
| 144 |
+
"params = model.to_fp16(params)"
|
|
|
|
|
|
|
| 145 |
],
|
| 146 |
"metadata": {
|
| 147 |
"id": "2r-zWmsPqTjh"
|
|
|
|
| 352 |
"# We can customize top_k/top_p used for generating samples\n",
|
| 353 |
"gen_top_k = None\n",
|
| 354 |
"gen_top_p = None\n",
|
| 355 |
+
"temperature = None\n",
|
| 356 |
"cond_scale = 3.0"
|
| 357 |
]
|
| 358 |
},
|