Merge pull request #18 from modelscope/v0.0.4_dev

V0.0.4 dev

Author: maochaojie

readme.md

@@ -176,7 +176,6 @@ python run.py --cfg classifier.yaml
 ```
 
 
-
 ## 🖥️ SCEPTER Studio
 
 ### Launch

scepter/modules/inference/diffusion_inference.py

@@ -395,7 +395,7 @@ def encode_first_stage(self, x, **kwargs):
             return self.first_stage_model['paras']['scale_factor'] * z
 
     def decode_first_stage(self, z):
-        _, dtype = self.get_function_info(self.first_stage_model, 'encode')
+        _, dtype = self.get_function_info(self.first_stage_model, 'decode')
         with torch.autocast('cuda',
                             enabled=dtype == 'float16',
                             dtype=getattr(torch, dtype)):