use recommended requirements and disable xformers

app.py

@@ -13,7 +13,7 @@ output_res = (768,768)
 
 conditioning_image_transforms = T.Compose(
     [
-        T2.ScaleJitter(target_size=output_res, scale_range=(0.5, 3.0)),
+        #T2.ScaleJitter(target_size=output_res, scale_range=(0.5, 3.0)),
         T.RandomCrop(size=output_res, pad_if_needed=True, padding_mode="symmetric"),
         T.ToTensor(),
         T.Normalize([0.5], [0.5]),
@@ -27,8 +27,8 @@ pipe = StableDiffusionControlNetPipeline.from_pretrained(
         torch_dtype=torch.float16,
         )
 pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
-pipe.enable_model_cpu_offload()
-pipe.enable_xformers_memory_efficient_attention()
+#pipe.enable_model_cpu_offload()
+#pipe.enable_xformers_memory_efficient_attention()
 
 generator = torch.manual_seed(0)
 

requirements.txt

@@ -1,13 +1,10 @@
 -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
-jax[cuda12_pip]
+jax[cuda11_pip]
 jaxlib
 git+https://github.com/huggingface/diffusers@main
 opencv-python
 transformers
 flax
 pillow
-torch==2.0.0
-triton
-torchvision==0.15.1
-git+https://github.com/huggingface/accelerate
-xformers
+torch
+torchvision