Spaces:
				
			
			
	
			
			
		Running
		
			on 
			
			A10G
	
	
	
			
			
	
	
	
	
		
		
		Running
		
			on 
			
			A10G
	Update app.py
Browse files
    	
        app.py
    CHANGED
    
    | @@ -89,7 +89,7 @@ with gr.Blocks(css="style.css") as demo: | |
| 89 | 
             
                    return image_editor, "Input has set!"
         | 
| 90 |  | 
| 91 |  | 
| 92 | 
            -
                 | 
| 93 | 
             
                def set_pipe(inversion_state, input_image, description_prompt, edit_guidance_scale, num_inference_steps=4,
         | 
| 94 | 
             
                             num_inversion_steps=4, inversion_max_step=0.6, rnri_iterations=2, rnri_alpha=0.1, rnri_lr=0.2):
         | 
| 95 | 
             
                    if device == 'cuda':
         | 
| @@ -129,7 +129,7 @@ with gr.Blocks(css="style.css") as demo: | |
| 129 | 
             
                    return image
         | 
| 130 |  | 
| 131 |  | 
| 132 | 
            -
                 | 
| 133 | 
             
                def edit(inversion_state, target_prompt):
         | 
| 134 | 
             
                    if inversion_state is None:
         | 
| 135 | 
             
                        raise gr.Error("Set inputs before editing.")
         | 
|  | |
| 89 | 
             
                    return image_editor, "Input has set!"
         | 
| 90 |  | 
| 91 |  | 
| 92 | 
            +
                @spaces.GPU
         | 
| 93 | 
             
                def set_pipe(inversion_state, input_image, description_prompt, edit_guidance_scale, num_inference_steps=4,
         | 
| 94 | 
             
                             num_inversion_steps=4, inversion_max_step=0.6, rnri_iterations=2, rnri_alpha=0.1, rnri_lr=0.2):
         | 
| 95 | 
             
                    if device == 'cuda':
         | 
|  | |
| 129 | 
             
                    return image
         | 
| 130 |  | 
| 131 |  | 
| 132 | 
            +
                @spaces.GPU
         | 
| 133 | 
             
                def edit(inversion_state, target_prompt):
         | 
| 134 | 
             
                    if inversion_state is None:
         | 
| 135 | 
             
                        raise gr.Error("Set inputs before editing.")
         |