"...git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "d46421446437511c931afd38ba3aa4908a00bdd9"
Unverified Commit bd2d2c75 authored by Da Zheng's avatar Da Zheng Committed by GitHub
Browse files

[BUGFIX] fix for MXNet 1.5. (#552)

* remove clone.

* turn on numpy compatible.

* Revert "remove clone."

This reverts commit 17bbf76ed72ff178df6b3f35addc428048672457.
parent 08d4900f
from __future__ import absolute_import
from distutils.version import LooseVersion
import numpy as np
import mxnet as mx
import mxnet.ndarray as nd
import numbers
MX_VERSION = LooseVersion(mx.__version__)
# After MXNet 1.5, empty tensors aren't supprted by default.
# after we turn on the numpy compatible flag, MXNet supports empty NDArray.
if MX_VERSION.version[0] == 1 and MX_VERSION.version[1] >= 5:
mx.set_np_compat(True)
def data_type_dict():
return {'float16' : np.float16,
'float32' : np.float32,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment