Update src/pixel3dmm/tracking/flame/FLAME.py

src/pixel3dmm/tracking/flame/FLAME.py

@@ -103,7 +103,7 @@ class FLAME(nn.Module):
         self._register_default_params('expression_params', config.num_exp_params)
 
         # Static and Dynamic Landmark embeddings for FLAME
-        lmk_embeddings = np.load(f'{env_paths.FLAME_ASSET}/FLAME2020/landmark_embedding.npy', allow_pickle=True, encoding='latin1')
+        lmk_embeddings = np.load(f'{env_paths.FLAME_MASK_ASSET}/FLAME2020/landmark_embedding.npy', allow_pickle=True, encoding='latin1')
         lmk_embeddings = lmk_embeddings[()]