| 
									
										
										
										
											2022-12-16 20:43:09 +08:00
										 |  |  | import sys | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-12-24 07:57:56 +03:00
										 |  |  | # this will break any attempt to import xformers which will prevent stability diffusion repo from trying to use it | 
					
						
							|  |  |  | if "--xformers" not in "".join(sys.argv): | 
					
						
							| 
									
										
										
										
											2022-12-24 11:17:21 +08:00
										 |  |  |     sys.modules["xformers"] = None | 
					
						
							| 
									
										
										
										
											2023-12-03 16:50:20 +02:00
										 |  |  | 
 | 
					
						
							|  |  |  | # Hack to fix a changed import in torchvision 0.17+, which otherwise breaks | 
					
						
							|  |  |  | # basicsr; see https://github.com/AUTOMATIC1111/stable-diffusion-webui/issues/13985 | 
					
						
							|  |  |  | try: | 
					
						
							|  |  |  |     import torchvision.transforms.functional_tensor  # noqa: F401 | 
					
						
							|  |  |  | except ImportError: | 
					
						
							|  |  |  |     try: | 
					
						
							|  |  |  |         import torchvision.transforms.functional as functional | 
					
						
							|  |  |  |         sys.modules["torchvision.transforms.functional_tensor"] = functional | 
					
						
							|  |  |  |     except ImportError: | 
					
						
							|  |  |  |         pass  # shrug... |