Requirements updates -testing

requirements.txt

@@ -17,7 +17,7 @@ Pillow
 numpy
 requests
 # git+https://github.com/huggingface/diffusers
-diffusers
+diffusers[torch]
 peft
 opencv-python
 open3d

utils/ai_generator_diffusers_flux.py

@@ -114,14 +114,14 @@ def generate_image_lowmem(
     strength=0.75,
     additional_parameters=None
 ):
-    # Retrieve the pipeline class from the mapping
-    print(f"device:{device}\nmodel_name:{model_name}\nlora_weights:{lora_weights}\n")
     print(f"\n {get_torch_info()}\n")
+    # Retrieve the pipeline class from the mapping
     pipeline_class = PIPELINE_CLASSES.get(pipeline_name)
     if not pipeline_class:
         raise ValueError(f"Unsupported pipeline type '{pipeline_name}'. "
                         f"Available options: {list(PIPELINE_CLASSES.keys())}")
     device = "cuda" if torch.cuda.is_available() else "cpu"
+    print(f"device:{device}\nmodel_name:{model_name}\nlora_weights:{lora_weights}\n")
     
     # Disable gradient calculations
     with torch.no_grad():

utils/constants.py

@@ -15,7 +15,7 @@ os.environ["PYTORCH_NVML_BASED_CUDA_CHECK"] = "1"
 os.environ["CUDA_VISIBLE_DEVICES"] = "0,1,2,3"
 
 # constants.py contains all the constants used in the project
-os.environ["HF_TOKEN"] = ""
+#os.environ["HF_TOKEN"] = ""
 HF_API_TOKEN = os.getenv("HF_TOKEN")
 default_lut_example_img = "./LUT/daisy.jpg"
 

utils/version_info.py

@@ -76,6 +76,8 @@ def versions_html():
          • 
         transformers: {get_transformers_version()}
          • 
+        xformers: {get_xformers_version()}
+         • 
         gradio: {gr.__version__}
          • 
         {toggle_dark_link}