Merge branch 'main' of https://huggingface.co/spaces/TIGER-Lab/GenAI-Arena

model/models/imagenhub_models.py

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

requirements.txt

@@ -42,10 +42,15 @@ madgrad>=1.1
 -e git+https://github.com/openai/CLIP.git@main#egg=clip
 dominate>=2.8.0
 -e git+https://github.com/CompVis/latent-diffusion.git#egg=latent-diffusion #ldm
+-e git+https://github.com/TIGER-AI-Lab/ImagenHub.git#egg=imagen-hub
 openai
 nltk~=3.8.1
 krippendorff
 statsmodels
+<<<<<<< HEAD
 plotly
 -e git+https://github.com/TIGER-AI-Lab/ImagenHub.git#egg=imagen-hub
-spaces
+spaces
+=======
+plotly
+>>>>>>> 01362d10b73735bae78449f871313391d9d21896