Update app.py

app.py

@@ -12,11 +12,9 @@ from safetensors import safe_open
 from compel import Compel, ReturnedEmbeddingsType
 from safetensors.torch import load_file
 
-model_id = 'aipicasso/emi'
+model_id = 'alfredplpl/emi-2-0'
 auth_token=os.environ["ACCESS_TOKEN"]
 
-scheduler = EulerAncestralDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler", use_auth_token=auth_token)
-
 pipe = StableDiffusionXLPipeline.from_pretrained(
   model_id,
   torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
@@ -26,7 +24,7 @@ pipe = StableDiffusionXLPipeline.from_pretrained(
 pipe=pipe.to("cuda")
 #pipe.enable_freeu(s1=0.9, s2=0.2, b1=1.3, b2=1.4)
 #pipe.enable_freeu(s1=0.6, s2=0.4, b1=1.1, b2=1.2)
-#pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
+pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
 
 state_dict = load_file("unaestheticXLv31.safetensors")
 pipe.load_textual_inversion(state_dict["clip_g"], token="unaestheticXLv31", text_encoder=pipe.text_encoder_2, tokenizer=pipe.tokenizer_2)
@@ -118,7 +116,7 @@ with gr.Blocks(css=css) as demo:
         f"""
             <div class="main-div">
               <div>
-                <h1>Emi Demo</h1>
+                <h1>DEmi Test</h1>
                 <!--
                 <h2>
                 Other Demos: 
@@ -130,7 +128,7 @@ with gr.Blocks(css=css) as demo:
                 -->
               </div>
               <p>
-               Demo for <a href="https://huggingface.co/aipicasso/emi">Emi</a><br>
+               Test for Distilled Emi<br>
               </p>
               <p>
               サンプル： そのままGenerateボタンを押してください。<br>
@@ -143,14 +141,8 @@ with gr.Blocks(css=css) as demo:
               sample prompt2 : 1man+, focus, wavy short hair, blue eyes, black shirt, white background, simple background
               </p>
               <p>
-              sample prompt3 : anime style, 1girl++
-              </p>
-              <p>
-              共有ボタンを押してみんなに画像を共有しましょう。Please push share button to share your image.
-              </p>
-              <p>
-              Running on {"<b>GPU 🔥</b>" if torch.cuda.is_available() else f"<b>CPU 🥶</b>. For faster inference it is recommended to <b>upgrade to GPU in <a href='https://huggingface.co/spaces/akhaliq/cool-japan-diffusion-2-1-0/settings'>Settings</a></b>"} <br>
               <a style="display:inline-block" href="https://huggingface.co/spaces/aipicasso/emi-latest-demo?duplicate=true"><img src="https://bit.ly/3gLdBN6" alt="Duplicate Space"></a> to say goodbye from waiting for the generating.
+              </p>
             </div>
         """
     )