Spaces:

baulab
/

ConceptSliders

Runtime error

RohitGandikota commited on Nov 28, 2023

Commit

b677cab

•

1 Parent(s): ec9ed03

pushing training code

Files changed (1) hide show

app.py CHANGED Viewed

@@ -229,7 +229,7 @@ class Demo:
         randn = torch.randint(1, 10000000, (1,)).item()
-        save_name = f'{randn}_{target_concept.replace(',','').replace(' ','').replace('.','')[:10]}_{positive_prompt.replace(',','').replace(' ','').replace('.','')[:10]}'
         save_name += f'_alpha-{1}'
         save_name += f'_noxattn'
         save_name += f'_rank_{rank}.pt'

         randn = torch.randint(1, 10000000, (1,)).item()
+        save_name = f"{randn}_{target_concept.replace(',','').replace(' ','').replace('.','')[:10]}_{positive_prompt.replace(',','').replace(' ','').replace('.','')[:10]}"
         save_name += f'_alpha-{1}'
         save_name += f'_noxattn'
         save_name += f'_rank_{rank}.pt'