Spaces:
				
			
			
	
			
			
		Paused
		
	
	
	
			
			
	
	
	
	
		
		
		Paused
		
	model update
Browse files
    	
        frontend/webui/hf_demo.py
    CHANGED
    
    | 
         @@ -37,13 +37,6 @@ def predict( 
     | 
|
| 37 | 
         
             
                seed,
         
     | 
| 38 | 
         
             
                use_seed,
         
     | 
| 39 | 
         
             
            ):
         
     | 
| 40 | 
         
            -
                lcm_text_to_image.init(
         
     | 
| 41 | 
         
            -
                    model_id=LCM_DEFAULT_MODEL_OPENVINO,
         
     | 
| 42 | 
         
            -
                    use_openvino=True,
         
     | 
| 43 | 
         
            -
                    use_lora=False,
         
     | 
| 44 | 
         
            -
                    lcm_lora=lcm_lora,
         
     | 
| 45 | 
         
            -
                    use_tiny_auto_encoder=False,
         
     | 
| 46 | 
         
            -
                )
         
     | 
| 47 | 
         
             
                print(f"prompt - {prompt}")
         
     | 
| 48 | 
         
             
                lcm_diffusion_setting = LCMDiffusionSetting()
         
     | 
| 49 | 
         
             
                lcm_diffusion_setting.prompt = prompt
         
     | 
| 
         @@ -56,6 +49,7 @@ def predict( 
     | 
|
| 56 | 
         
             
                lcm_diffusion_setting.image_width = 320 if is_openvino_device() else 512
         
     | 
| 57 | 
         
             
                lcm_diffusion_setting.image_height = 320 if is_openvino_device() else 512
         
     | 
| 58 | 
         
             
                lcm_diffusion_setting.use_openvino = True if is_openvino_device() else False
         
     | 
| 
         | 
|
| 59 | 
         
             
                start = perf_counter()
         
     | 
| 60 | 
         
             
                images = lcm_text_to_image.generate(lcm_diffusion_setting)
         
     | 
| 61 | 
         
             
                latency = perf_counter() - start
         
     | 
| 
         | 
|
| 37 | 
         
             
                seed,
         
     | 
| 38 | 
         
             
                use_seed,
         
     | 
| 39 | 
         
             
            ):
         
     | 
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 
         | 
|
| 40 | 
         
             
                print(f"prompt - {prompt}")
         
     | 
| 41 | 
         
             
                lcm_diffusion_setting = LCMDiffusionSetting()
         
     | 
| 42 | 
         
             
                lcm_diffusion_setting.prompt = prompt
         
     | 
| 
         | 
|
| 49 | 
         
             
                lcm_diffusion_setting.image_width = 320 if is_openvino_device() else 512
         
     | 
| 50 | 
         
             
                lcm_diffusion_setting.image_height = 320 if is_openvino_device() else 512
         
     | 
| 51 | 
         
             
                lcm_diffusion_setting.use_openvino = True if is_openvino_device() else False
         
     | 
| 52 | 
         
            +
                lcm_text_to_image.init(lcm_diffusion_setting)
         
     | 
| 53 | 
         
             
                start = perf_counter()
         
     | 
| 54 | 
         
             
                images = lcm_text_to_image.generate(lcm_diffusion_setting)
         
     | 
| 55 | 
         
             
                latency = perf_counter() - start
         
     |