Update app.py
Browse files
app.py
CHANGED
@@ -16,7 +16,7 @@ if not torch.cuda.is_available():
|
|
16 |
DESCRIPTION += "\n<p>Running on CPU 🥶 This demo may not work on CPU.</p>"
|
17 |
|
18 |
MAX_SEED = np.iinfo(np.int32).max
|
19 |
-
CACHE_EXAMPLES = torch.cuda.is_available() and os.getenv("CACHE_EXAMPLES", "1") == "
|
20 |
MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "1536"))
|
21 |
USE_TORCH_COMPILE = os.getenv("USE_TORCH_COMPILE", "0") == "1"
|
22 |
ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD", "0") == "1"
|
@@ -56,7 +56,7 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
|
|
56 |
return seed
|
57 |
|
58 |
|
59 |
-
@spaces.GPU
|
60 |
def generate(
|
61 |
prompt: str,
|
62 |
negative_prompt: str = "",
|
@@ -64,7 +64,7 @@ def generate(
|
|
64 |
seed: int = 0,
|
65 |
width: int = 1024,
|
66 |
height: int = 1024,
|
67 |
-
guidance_scale: float =
|
68 |
randomize_seed: bool = False,
|
69 |
use_resolution_binning: bool = True,
|
70 |
progress=gr.Progress(track_tqdm=True),
|
@@ -96,10 +96,6 @@ def generate(
|
|
96 |
|
97 |
examples = [
|
98 |
"neon holography crystal cat",
|
99 |
-
"a cat eating a piece of cheese",
|
100 |
-
"an astronaut riding a horse in space",
|
101 |
-
"a cartoon of a boy playing with a tiger",
|
102 |
-
"a cute robot artist painting on an easel, concept art",
|
103 |
"a close up of a woman wearing a transparent, prismatic, elaborate nemeses headdress, over the should pose, brown skin-tone"
|
104 |
]
|
105 |
|
|
|
16 |
DESCRIPTION += "\n<p>Running on CPU 🥶 This demo may not work on CPU.</p>"
|
17 |
|
18 |
MAX_SEED = np.iinfo(np.int32).max
|
19 |
+
CACHE_EXAMPLES = torch.cuda.is_available() and os.getenv("CACHE_EXAMPLES", "1") == "0"
|
20 |
MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "1536"))
|
21 |
USE_TORCH_COMPILE = os.getenv("USE_TORCH_COMPILE", "0") == "1"
|
22 |
ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD", "0") == "1"
|
|
|
56 |
return seed
|
57 |
|
58 |
|
59 |
+
@spaces.GPU
|
60 |
def generate(
|
61 |
prompt: str,
|
62 |
negative_prompt: str = "",
|
|
|
64 |
seed: int = 0,
|
65 |
width: int = 1024,
|
66 |
height: int = 1024,
|
67 |
+
guidance_scale: float = 5,
|
68 |
randomize_seed: bool = False,
|
69 |
use_resolution_binning: bool = True,
|
70 |
progress=gr.Progress(track_tqdm=True),
|
|
|
96 |
|
97 |
examples = [
|
98 |
"neon holography crystal cat",
|
|
|
|
|
|
|
|
|
99 |
"a close up of a woman wearing a transparent, prismatic, elaborate nemeses headdress, over the should pose, brown skin-tone"
|
100 |
]
|
101 |
|