wassname/denoising-diffusion-pytorch
1
0
Fork 0
mirror of https://github.com/wassname/denoising-diffusion-pytorch.git synced 2026-09-09 11:21:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

allow for mixed precision training with fp16 flag

Browse Source
This commit is contained in:
Phil Wang
2020-09-08 17:25:26 -07:00
parent 88f83d0ff2
commit b72235e71d
3 changed files with 50 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
setup.py
+2 -2
View File
@@ -3,7 +3,7 @@ from setuptools import setup, find_packages
setup(
name = 'denoising-diffusion-pytorch',
packages = find_packages(),
version = '0.2.2',
version = '0.2.3',
license='MIT',
description = 'Denoising Diffusion Probabilistic Models - Pytorch',
author = 'Phil Wang',
@@ -17,7 +17,7 @@ setup(
'einops',
'numpy',
'pillow',
'torch',
'torch>=1.6',
'torchvision',
'tqdm'
],