已合并
bugfix #207
guowenna1创建于 3月13日
bugfix #207
已合并
共 1 个文件变更+4-4
| @@ -27,6 +27,10 @@ __all__ = [ | |||
| 27 | 'fast_layernorm' | 27 | 'fast_layernorm' |
| 28 | ] | 28 | ] |
| 29 | 29 | ||
| 30 | + | ||
| 31 | +from .env import set_environment_variables | ||
| 32 | +set_environment_variables() | ||
| 33 | + | ||
| 30 | from .cache_agent import CacheConfig, CacheAgent | 34 | from .cache_agent import CacheConfig, CacheAgent |
| 31 | from .layers import ( | 35 | from .layers import ( |
| 32 | layernorm_scale_shift, | 36 | layernorm_scale_shift, |
| @@ -39,7 +43,3 @@ from .layers import ( | |||
| 39 | fast_layernorm | 43 | fast_layernorm |
| 40 | ) | 44 | ) |
| 41 | from .quantization import quantize, TimestepManager, TimestepPolicyConfig | 45 | from .quantization import quantize, TimestepManager, TimestepPolicyConfig |
| 42 | -from .env import set_environment_variables | ||
| 43 | - | ||
| 44 | - | ||
| 45 | -set_environment_variables() | ||