Spaces:

teticio
/

audio-diffusion

Running

teticio commited on Dec 30, 2022

Commit

80c2c54

•

2 Parent(s): 65031fe 5a187d9

Merge pull request #21 from nturusin/fix/train_model_nb_updated_command

Files changed (1) hide show

notebooks/train_model.ipynb CHANGED Viewed

@@ -133,7 +133,7 @@
    },
    "outputs": [],
    "source": [
-    "!python scripts/train_unconditional.py \\\n",
     "  --dataset_name data \\\n",
     "  --output_dir model \\\n",
     "  --num_epochs 10 \\\n",

    },
    "outputs": [],
    "source": [
+    "!python scripts/train_unet.py \\\n",
     "  --dataset_name data \\\n",
     "  --output_dir model \\\n",
     "  --num_epochs 10 \\\n",