Skip to content

Commit

Permalink
improve base image model
Browse files Browse the repository at this point in the history
  • Loading branch information
surcyf123 authored Jan 30, 2024
1 parent 4db65f8 commit ab8ebeb
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions validators/image_validator.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,10 +21,10 @@ def __init__(self, dendrite, config, subtensor, wallet):
super().__init__(dendrite, config, subtensor, wallet, timeout=25)
self.streaming = False
self.query_type = "images"
self.model = "dall-e-2"
self.model = "dall-e-3"
self.weight = .5
self.provider = "OpenAI"
self.size = "1024x1024"
self.size = "1792x1024"
self.width = 1024
self.height = 1024
self.quality = "standard"
Expand All @@ -46,15 +46,15 @@ async def start_query(self, available_uids, metagraph):
uid_to_question = {}

# Randomly choose the provider based on specified probabilities
providers = ["OpenAI"] * 6 + ["Stability"] * 4
providers = ["OpenAI"] * 8 + ["Stability"] * 2
self.provider = random.choice(providers)

if self.provider == "Stability":
self.seed = random.randint(1000, 1000000)
self.model = "stable-diffusion-xl-1024-v1-0"

elif self.provider == "OpenAI":
self.model = "dall-e-2"
self.model = "dall-e-3"

# Query all images concurrently
for uid in available_uids:
Expand Down

0 comments on commit ab8ebeb

Please sign in to comment.