Minor change to falcon pipeline

This commit is contained in:
Vivek Khandelwal
2023-06-19 17:03:45 +00:00
parent 015d0132c3
commit fafd713141

View File

@@ -99,7 +99,7 @@ class Falcon(SharkLLMBase):
return falcon_model
def compile_falcon(self):
if args.use_precompiled_vmfb:
if args.use_precompiled_model:
if not self.falcon_vmfb_path.exists():
# Downloading VMFB from shark_tank
download_public_file(