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

V0.0.4 dev
modelscope · Apr 10, 2024 · 0010d42 · 0010d42
2 parents 36b259b + 8a9b791
commit 0010d42
Show file tree

Hide file tree

Showing 5 changed files with 25 additions and 1,072 deletions.
diff --git a/readme.md b/readme.md
@@ -176,7 +176,6 @@ python run.py --cfg classifier.yaml
 ```
 
 
-
 ## 🖥️ SCEPTER Studio
 
 ### Launch

diff --git a/scepter/modules/inference/diffusion_inference.py b/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)):