Spaces:
Runtime error
Runtime error
YongLD
commited on
Commit
•
71cefe5
1
Parent(s):
8a3ba61
add example
Browse files
app.py
CHANGED
@@ -19,7 +19,7 @@ def filename_from_text(text: str) -> str:
|
|
19 |
def log_gpu_memory():
|
20 |
print(subprocess.check_output('nvidia-smi').decode('utf-8'))
|
21 |
|
22 |
-
|
23 |
|
24 |
model = MinDalle(
|
25 |
is_mega=True,
|
@@ -28,7 +28,7 @@ model = MinDalle(
|
|
28 |
dtype=torch.float32
|
29 |
)
|
30 |
|
31 |
-
|
32 |
|
33 |
def run_model(
|
34 |
text: str,
|
@@ -83,7 +83,7 @@ def run_model(
|
|
83 |
is_verbose = True
|
84 |
)
|
85 |
|
86 |
-
|
87 |
|
88 |
ext = 'png' if bool(save_as_png) else 'jpg'
|
89 |
filename = filename_from_text(text)
|
@@ -161,6 +161,9 @@ with demo:
|
|
161 |
examples=[
|
162 |
['A white cat with golden sunglasses on, pink background, studio lighting, 4k, award winning photography', 2, 'examples/cat.png'],
|
163 |
['an astronaut dancing on the moon’s surface, close-up photo', 2, 'examples/astronaut.png'],
|
|
|
|
|
|
|
164 |
],
|
165 |
inputs=[
|
166 |
input_text,
|
|
|
19 |
def log_gpu_memory():
|
20 |
print(subprocess.check_output('nvidia-smi').decode('utf-8'))
|
21 |
|
22 |
+
log_gpu_memory()
|
23 |
|
24 |
model = MinDalle(
|
25 |
is_mega=True,
|
|
|
28 |
dtype=torch.float32
|
29 |
)
|
30 |
|
31 |
+
log_gpu_memory()
|
32 |
|
33 |
def run_model(
|
34 |
text: str,
|
|
|
83 |
is_verbose = True
|
84 |
)
|
85 |
|
86 |
+
log_gpu_memory()
|
87 |
|
88 |
ext = 'png' if bool(save_as_png) else 'jpg'
|
89 |
filename = filename_from_text(text)
|
|
|
161 |
examples=[
|
162 |
['A white cat with golden sunglasses on, pink background, studio lighting, 4k, award winning photography', 2, 'examples/cat.png'],
|
163 |
['an astronaut dancing on the moon’s surface, close-up photo', 2, 'examples/astronaut.png'],
|
164 |
+
['A photo of a Samoyed dog with its tongue out hugging a white Siamese cat', 5, 'examples/dog.png'],
|
165 |
+
['Dragons of Earth, Wind, Fire, powering up a huge sphere of compressed energy, digital art', 2, 'examples/dragon.png'],
|
166 |
+
['A snowboarder jumping in the air while coming down a ski mountain, concept art, artstation, unreal engine, 3d render, HD, Bokeh', 3, 'examples/snow.png'],
|
167 |
],
|
168 |
inputs=[
|
169 |
input_text,
|