fluidflow 0.2.1__tar.gz → 0.2.2__tar.gz
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- {fluidflow-0.2.1 → fluidflow-0.2.2}/PKG-INFO +1 -1
- {fluidflow-0.2.1 → fluidflow-0.2.2}/fluidFlow/trainer.py +1 -2
- {fluidflow-0.2.1 → fluidflow-0.2.2}/fluidflow.egg-info/PKG-INFO +1 -1
- {fluidflow-0.2.1 → fluidflow-0.2.2}/pyproject.toml +1 -1
- {fluidflow-0.2.1 → fluidflow-0.2.2}/LICENSE +0 -0
- {fluidflow-0.2.1 → fluidflow-0.2.2}/README.md +0 -0
- {fluidflow-0.2.1 → fluidflow-0.2.2}/fluidFlow/__init__.py +0 -0
- {fluidflow-0.2.1 → fluidflow-0.2.2}/fluidFlow/attention.py +0 -0
- {fluidflow-0.2.1 → fluidflow-0.2.2}/fluidFlow/basic_modules.py +0 -0
- {fluidflow-0.2.1 → fluidflow-0.2.2}/fluidFlow/dit.py +0 -0
- {fluidflow-0.2.1 → fluidflow-0.2.2}/fluidFlow/flow_matching/__init__.py +0 -0
- {fluidflow-0.2.1 → fluidflow-0.2.2}/fluidFlow/flow_matching/integrators.py +0 -0
- {fluidflow-0.2.1 → fluidflow-0.2.2}/fluidFlow/flow_matching/path.py +0 -0
- {fluidflow-0.2.1 → fluidflow-0.2.2}/fluidFlow/flow_matching/transport.py +0 -0
- {fluidflow-0.2.1 → fluidflow-0.2.2}/fluidFlow/moe.py +0 -0
- {fluidflow-0.2.1 → fluidflow-0.2.2}/fluidFlow/unet.py +0 -0
- {fluidflow-0.2.1 → fluidflow-0.2.2}/fluidflow.egg-info/SOURCES.txt +0 -0
- {fluidflow-0.2.1 → fluidflow-0.2.2}/fluidflow.egg-info/dependency_links.txt +0 -0
- {fluidflow-0.2.1 → fluidflow-0.2.2}/fluidflow.egg-info/requires.txt +0 -0
- {fluidflow-0.2.1 → fluidflow-0.2.2}/fluidflow.egg-info/top_level.txt +0 -0
- {fluidflow-0.2.1 → fluidflow-0.2.2}/setup.cfg +0 -0
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
|
|
1
|
+
from pathlib import Path
|
|
2
2
|
|
|
3
3
|
from matplotlib import pyplot as plt
|
|
4
4
|
import numpy as np
|
|
@@ -10,7 +10,6 @@ from torch.profiler import profile, ProfilerActivity, schedule
|
|
|
10
10
|
|
|
11
11
|
from accelerate import Accelerator, DataLoaderConfiguration
|
|
12
12
|
from accelerate import FullyShardedDataParallelPlugin
|
|
13
|
-
from torch.distributed.fsdp import ShardingStrategy
|
|
14
13
|
from torch.distributed.fsdp import MixedPrecisionPolicy
|
|
15
14
|
from ema_pytorch import EMA
|
|
16
15
|
|
|
@@ -5,7 +5,7 @@ build-backend = "setuptools.build_meta"
|
|
|
5
5
|
[project]
|
|
6
6
|
name = "fluidflow"
|
|
7
7
|
authors = [{ name = "David Ramos", email = "david.ramos.archilla@upm.es" }]
|
|
8
|
-
version = "0.2.
|
|
8
|
+
version = "0.2.2"
|
|
9
9
|
description = "FluidFlow: models and training utilities for flow matching and diffusion"
|
|
10
10
|
readme = "README.md"
|
|
11
11
|
requires-python = ">=3.8"
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|