Spaces:

baulab
/

ConceptSliders

Sleeping

RohitGandikota commited on Nov 28, 2023

Commit

675f687

1 Parent(s): f83a5f5

pushing training code

Files changed (1) hide show

app.py CHANGED Viewed

@@ -278,7 +278,7 @@ class Demo:
         name = os.path.basename(model_path)
         rank = 4
         alpha = 1
-        if rank in model_path:
             rank = int(model_path.split('_')[-1].replace('.pt',''))
 #         if 'rank4' in model_path:
 #             rank = 4

         name = os.path.basename(model_path)
         rank = 4
         alpha = 1
+        if 'rank' in model_path:
             rank = int(model_path.split('_')[-1].replace('.pt',''))
 #         if 'rank4' in model_path:
 #             rank = 4