Skip to content

H
Hydra Node Http
Commit 898ffdc2 authored by novelailab's avatar novelailab
Browse files
Options

grid_size

parent deea3042
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
hydra_node/models.py
View file @ 898ffdc2
...@@ -140,7 +140,7 @@ class DalleMiniModel(nn.Module): ...@@ -140,7 +140,7 @@ class DalleMiniModel(nn.Module):
images = self.model.generate_images( images = self.model.generate_images(
text=request.prompt, text=request.prompt,
seed=seed, seed=seed,
grid_size=4, grid_size=request.grid_size,
is_seamless=False, is_seamless=False,
temperature=request.temp, temperature=request.temp,
top_k=request.top_k, top_k=request.top_k,
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment