Merge pull request #17 from acceleratescience/feature/fg-updates

Updates made to slides for diffusion models 03/06

Author: fin-griffin

.gitignore

@@ -13,4 +13,12 @@ __pycache__
 .idea/
 
 # Python package build artifacts
-*.egg-info/
+*.egg-info/
+
+notes.md
+
+# Office lock files
+~$*.pptx
+
+# Pre-intro slides (these are very large)
+slides/pre_intro/

.pre-commit-config.yaml

@@ -1,4 +1,8 @@
 repos:
+  - repo: https://github.com/kynan/nbstripout
+    rev: 0.8.1
+    hooks:
+      - id: nbstripout
   - repo: https://github.com/astral-sh/ruff-pre-commit
     # Ruff version.
     rev: v0.11.7

notebooks/1_VAEs.ipynb

@@ -16,6 +16,8 @@ dependencies = [
     "transformers==4.41.2",
     "scikit-learn>=1.5.2",
     "ipykernel>=6.30.1",
+    "torch>=2.12.0",
+    "torchvision>=0.27.0",
 ]
 
 [dependency-groups]
@@ -33,4 +35,4 @@ packages = ["diffusion_models"]
 
 [build-system]
 requires = ["setuptools>=61.0"]
-build-backend = "setuptools.build_meta"
+build-backend = "setuptools.build_meta"