mrleo1nid/ACE-Step
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix environ check

Browse Source
This commit is contained in:
Michael Hedman
2025-05-16 18:10:26 +02:00
parent 42884d94b7
commit 79dd277e21
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
acestep/pipeline_ace_step.py
+1 -1
View File
@@ -129,7 +129,7 @@ class ACEStepPipeline:
self.dtype = torch.float16 self.dtype = torch.float16
if device.type == "mps": if device.type == "mps":
self.dtype = torch.float32 self.dtype = torch.float32
if len(os.environ['COLAB_FREE']): if 'COLAB_FREE' in os.environ and len(os.environ['COLAB_FREE']):
self.dtype = torch.float16 self.dtype = torch.float16
self.device = device self.device = device
self.loaded = False self.loaded = False