diff --git a/app/app_sana.py b/app/app_sana.py index a392924..4efd71d 100755 --- a/app/app_sana.py +++ b/app/app_sana.py @@ -21,6 +21,7 @@ import random import time import uuid +import socket from datetime import datetime import gradio as gr @@ -308,7 +309,7 @@ def generate(
Sana-{model_size}B{args.image_size}px
Sana: Efficient High-Resolution Image Synthesis with Linear Diffusion Transformer
[Paper] [Github(coming soon)] [Project] -
Powered by DC-AE with 32x latent space
, running on A6000 node. +Powered by DC-AE with 32x latent space,
running on node {socket.gethostname()}.Unsafe word will give you a 'Red Heart' in the image instead.
""" if model_size == "0.6": @@ -336,7 +337,7 @@ def generate( """ with gr.Blocks(css=css) as demo: gr.Markdown(title) - gr.Markdown(DESCRIPTION) + gr.HTML(DESCRIPTION) gr.DuplicateButton( value="Duplicate Space for private use", elem_id="duplicate-button", @@ -485,4 +486,4 @@ def generate( ) if __name__ == "__main__": - demo.queue(max_size=20).launch(server_name="0.0.0.0", server_port=DEMO_PORT, debug=True, share=True) + demo.queue(max_size=20).launch(server_name="0.0.0.0", server_port=DEMO_PORT, debug=False, share=False)