Spaces:
				
			
			
	
			
			
		Running
		
			on 
			
			Zero
	
	
	
			
			
	
	
	
	
		
		
		Running
		
			on 
			
			Zero
	Update app_with_diffusers.py
#15
by
						
p20523810
	
							
						- opened
							
					
- app_with_diffusers.py +2 -2
    	
        app_with_diffusers.py
    CHANGED
    
    | @@ -40,7 +40,7 @@ def resize_img(input_image, max_side=1280, min_side=1024, size=None, | |
| 40 | 
             
            instantir_path = f'./models'
         | 
| 41 |  | 
| 42 | 
             
            # load pretrained models
         | 
| 43 | 
            -
            pipe = InstantIRPipeline.from_pretrained('stabilityai/stable-diffusion-xl-base- | 
| 44 |  | 
| 45 | 
             
            # load adapter
         | 
| 46 | 
             
            load_adapter_to_pipe(
         | 
| @@ -51,7 +51,7 @@ load_adapter_to_pipe( | |
| 51 |  | 
| 52 | 
             
            # load previewer lora
         | 
| 53 | 
             
            pipe.prepare_previewers(instantir_path)
         | 
| 54 | 
            -
            pipe.scheduler = DDPMScheduler.from_pretrained('stabilityai/stable-diffusion-xl-base- | 
| 55 | 
             
            lcm_scheduler = LCMSingleStepScheduler.from_config(pipe.scheduler.config)
         | 
| 56 |  | 
| 57 | 
             
            # load aggregator weights
         | 
|  | |
| 40 | 
             
            instantir_path = f'./models'
         | 
| 41 |  | 
| 42 | 
             
            # load pretrained models
         | 
| 43 | 
            +
            pipe = InstantIRPipeline.from_pretrained('stabilityai/stable-diffusion-xl-base-0.9', torch_dtype=torch.float16)
         | 
| 44 |  | 
| 45 | 
             
            # load adapter
         | 
| 46 | 
             
            load_adapter_to_pipe(
         | 
|  | |
| 51 |  | 
| 52 | 
             
            # load previewer lora
         | 
| 53 | 
             
            pipe.prepare_previewers(instantir_path)
         | 
| 54 | 
            +
            pipe.scheduler = DDPMScheduler.from_pretrained('stabilityai/stable-diffusion-xl-base-0.9', subfolder="scheduler")
         | 
| 55 | 
             
            lcm_scheduler = LCMSingleStepScheduler.from_config(pipe.scheduler.config)
         | 
| 56 |  | 
| 57 | 
             
            # load aggregator weights
         | 
