Spaces:

TIGER-Lab
/

GenAI-Arena

Running on Zero

tianleliphoebe commited on Mar 31, 2024

Commit

0906566

verified ·

1 Parent(s): a8ec059

Update model/models/imagenhub_models.py

Files changed (1) hide show

model/models/imagenhub_models.py CHANGED Viewed

@@ -22,7 +22,7 @@ class Prompt2prompt(ImagenHubModel):
     def __call__(self, *args, **kwargs):
         if "num_inner_steps" not in kwargs:
-            kwargs["num_inner_steps"] = 2
         return super().__call__(*args, **kwargs)
 def load_imagenhub_model(model_name, model_type=None):

     def __call__(self, *args, **kwargs):
         if "num_inner_steps" not in kwargs:
+            kwargs["num_inner_steps"] = 3
         return super().__call__(*args, **kwargs)
 def load_imagenhub_model(model_name, model_type=None):