update

hf_demo.py

@@ -46,7 +46,7 @@ def demo_inference_gen(adapter_choice:str, prompt:str, samples:int=1,seed:int=0,
                             height=512, width=512, scales=[1.0],
                             save_dir=None, seed=seed,steps=steps, guidance_scale=guidance_scale,
                             start_noise=-1, show=False, style_prompt="sks art", no_load=True,
-                            from_scratch=True)[0][1.0]
+                            from_scratch=True, device=device)[0][1.0]
     return pred_images
 
 def demo_inference_stylization(adapter_path:str, prompts:list, image:list, start_noise=800,seed:int=0):
@@ -56,7 +56,7 @@ def demo_inference_stylization(adapter_path:str, prompts:list, image:list, start
                             height=512, width=512, scales=[0.,1.],
                             save_dir=None, seed=seed,steps=20, guidance_scale=7.5,
                             start_noise=start_noise, show=True, style_prompt="sks art", no_load=True,
-                            from_scratch=False)
+                            from_scratch=False, device=device)
     return pred_images
 
 # def infer(prompt, samples, steps, scale, seed):

hf_demo_test.ipynb

@@ -241,6 +241,7 @@
     "# Direct infer\n",
     "with block:\n",
     "    with gr.Group():\n",
+    "        gr.Markdown(\" # Art-Free Diffusion Demo\")\n",
     "        with gr.Row():\n",
     "            text = gr.Textbox(\n",
     "                label=\"Enter your prompt\",\n",