Update raw.py

raw.py

@@ -44,37 +44,6 @@ pipe = FluxControlNetPipeline.from_pretrained(
 # adapter_id3 = "enhanceaiteam/Flux-uncensored-v2"
 
 pipe.to("cuda")
-try:
-    apply_group_offloading(
-        pipe.transformer,
-        offload_type="leaf_level",
-        offload_device=torch.device("cpu"),
-        onload_device=torch.device("cuda"),
-        use_stream=True,
-    )
-    apply_group_offloading(
-        pipe.text_encoder, 
-        offload_device=torch.device("cpu"),
-        onload_device=torch.device("cuda"),
-        offload_type="leaf_level",
-        use_stream=True,
-    )
-    apply_group_offloading(
-        pipe.text_encoder_2, 
-        offload_device=torch.device("cpu"),
-        onload_device=torch.device("cuda"),
-        offload_type="leaf_level",
-        use_stream=True,
-    )
-    apply_group_offloading(
-        pipe.vae, 
-        offload_device=torch.device("cpu"),
-        onload_device=torch.device("cuda"),
-        offload_type="leaf_level",
-        use_stream=True,
-    )
-except:
-    print("debug-group")
 
 try:
     pipe.enable_sequential_cpu_offload()
@@ -88,11 +57,6 @@ try:
     pipe.vae.enable_tiling()
 except:
     print("debug-3")
-try:
-    pipe.enable_xformers_memory_efficient_attention()
-except:
-    print("debug-4")
-
 
 # pipe.load_lora_weights(adapter_id, adapter_name="turbo")
 # pipe.load_lora_weights(adapter_id2, adapter_name="real")