text-to-image

Paused

Zaiiida commited on Dec 2, 2024

Commit

571bfa0

verified ·

1 Parent(s): 369d95b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,7 +17,7 @@ else:
 pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
 pipe = pipe.to(device)
-MAX_IMAGE_SIZE = 512
 examples = [
     "A border collie lying in some Fall leaves as the forest trees change colors",
@@ -29,8 +29,8 @@ examples = [
 @spaces.GPU(duration=30)
 def infer(
     prompt,
-    width=512,
-    height=512,
     num_inference_steps=20,
     guidance_scale=7.5,
 ):

 pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
 pipe = pipe.to(device)
+MAX_IMAGE_SIZE = 1024
 examples = [
     "A border collie lying in some Fall leaves as the forest trees change colors",
 @spaces.GPU(duration=30)
 def infer(
     prompt,
+    width=1024,
+    height=1024,
     num_inference_steps=20,
     guidance_scale=7.5,
 ):