nodes/kijai-ComfyUI-DynamiCrafterWrapper
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Default to fp16 unet with 'auto'

Browse Source
This commit is contained in:
Kijai
2024-03-19 16:28:42 +02:00
parent 6a9ad93186
commit c6178d9fdc
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes.py
+3 -3
View File
@@ -83,10 +83,10 @@ class DynamiCrafterModelLoader:
self.model.eval() self.model.eval()
if dtype == "auto": if dtype == "auto":
try: try:
if mm.should_use_bf16(): if mm.should_use_fp16():
self.model.to(convert_dtype('bf16'))
elif mm.should_use_fp16():
self.model.to(convert_dtype('fp16')) self.model.to(convert_dtype('fp16'))
elif mm.should_use_bf16():
self.model.to(convert_dtype('bf16'))
else: else:
self.model.to(convert_dtype('fp32')) self.model.to(convert_dtype('fp32'))
except: except: