diff --git a/configs/config_abl_overfit_jumpsuit_2.py b/configs/config_abl_overfit_jumpsuit_2.py
index 2125f42..86feac6 100644
--- a/configs/config_abl_overfit_jumpsuit_2.py
+++ b/configs/config_abl_overfit_jumpsuit_2.py
@@ -19,8 +19,8 @@ config['only_curvatures'] =           True
 ablation_count = [config['use_skinning'], config['remove_global_orientation'], config['use_curvature'], config['use_extra_losses'], config['use_offsets']].count(False)
 assert ablation_count == 1 or ablation_count == 0
 
-config['gpu'] = '5'
-config['name'] = 'abl_overfit_jumpsuit_2'
+config['gpu'] = '3'
+config['name'] = 'abl_overfit_jumpsuit_2_r4'
 config['notes'] = 'remove other conditions from input as well'
 
 config['avg_garment_path'] = 'custom_data/train_jumpsuit_30k_no_shape_avg_uv.npy' #custom_data/train_no_shape_avg_uv.npy # train_no_shape_avg_uv  train_no_shape_one_hot_avg_uv
diff --git a/cvae.py b/cvae.py
index 59d7c75..a2de40f 100644
--- a/cvae.py
+++ b/cvae.py
@@ -1,6 +1,6 @@
 import os
 import sys
-from configs.config_abl_all_post import config
+from configs.config_abl_overfit_jumpsuit_2 import config
 
 os.environ["CUDA_VISIBLE_DEVICES"] = config['gpu']
 import tensorflow as tf
@@ -41,7 +41,7 @@ from losses import collision_loss as collision_penalty
 from custom_utils.custom_utils import batch_to_dict, load_model, plot_3d, UV2Mesh, get_summary_html, maskFacesBatch, _maskFaces, plot_3d_collision
 from dataset import DatasetFactory, StaticDataset
 from custom_utils.smpl import SMPL
-from models.cvae_model import CVAENew as CVAE
+from models.cvae_model import CVAE
 
 
 def init_wandb():