Spaces:
				
			
			
	
			
			
		Runtime error
		
	
	
	
			
			
	
	
	
	
		
		
		Runtime error
		
	Commit 
							
							·
						
						18e5bb5
	
1
								Parent(s):
							
							a0466aa
								
Enable scheduler in DiffusionPipeline and adjust default slider value for steps
Browse files
    	
        app.py
    CHANGED
    
    | @@ -87,7 +87,7 @@ scheduler_config = { | |
| 87 | 
             
            scheduler = FlowMatchEulerDiscreteScheduler.from_config(scheduler_config)
         | 
| 88 | 
             
            pipe = DiffusionPipeline.from_pretrained(
         | 
| 89 | 
             
                base_model, 
         | 
| 90 | 
            -
                 | 
| 91 | 
             
                torch_dtype=dtype
         | 
| 92 | 
             
            ).to(device)
         | 
| 93 |  | 
| @@ -458,7 +458,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css, delete_cache=(60, 60)) as app: | |
| 458 | 
             
                                    minimum=1.0, 
         | 
| 459 | 
             
                                    maximum=5.0, 
         | 
| 460 | 
             
                                    step=0.1, 
         | 
| 461 | 
            -
                                    value= | 
| 462 | 
             
                                    info="Lower for speed mode, higher for quality"
         | 
| 463 | 
             
                                )
         | 
| 464 | 
             
                                steps = gr.Slider(
         | 
|  | |
| 87 | 
             
            scheduler = FlowMatchEulerDiscreteScheduler.from_config(scheduler_config)
         | 
| 88 | 
             
            pipe = DiffusionPipeline.from_pretrained(
         | 
| 89 | 
             
                base_model, 
         | 
| 90 | 
            +
                scheduler=scheduler, 
         | 
| 91 | 
             
                torch_dtype=dtype
         | 
| 92 | 
             
            ).to(device)
         | 
| 93 |  | 
|  | |
| 458 | 
             
                                    minimum=1.0, 
         | 
| 459 | 
             
                                    maximum=5.0, 
         | 
| 460 | 
             
                                    step=0.1, 
         | 
| 461 | 
            +
                                    value=1,
         | 
| 462 | 
             
                                    info="Lower for speed mode, higher for quality"
         | 
| 463 | 
             
                                )
         | 
| 464 | 
             
                                steps = gr.Slider(
         | 
 
			

