Update raw.py

raw.py

@@ -7,11 +7,11 @@ from diffusers import BitsAndBytesConfig as DiffusersBitsAndBytesConfig
 from transformers import T5EncoderModel
 from transformers import BitsAndBytesConfig as TransformersBitsAndBytesConfig
 from peft import PeftModel, PeftConfig
-from attention_map_diffusers import (
-    attn_maps,
-    init_pipeline,
-    save_attention_maps
-)
+# from attention_map_diffusers import (
+#     attn_maps,
+#     init_pipeline,
+#     save_attention_maps
+# )
 import gradio as gr
 huggingface_token = os.getenv("HUGGINFACE_TOKEN")
 
@@ -50,7 +50,6 @@ pipe.load_lora_weights(adapter_id3, weight_name="lora.safetensors", adapter_name
 pipe.set_adapters(["turbo", "real", "enhance"], adapter_weights=[0.9, 0.66, 0.6])
 pipe.fuse_lora(adapter_names=["turbo", "real", "enhance"], lora_scale=1.0)
 pipe.unload_lora_weights()
-pipe = init_pipeline(pipe)
 # pipe.enable_xformers_memory_efficient_attention()
 # save to the Hub
 # pipe.push_to_hub("fused-t-r")