Merge branch 'main' into new_mame

updating the main changes in the main branch, such as environmental changes and citation

CITATION.cff

@@ -0,0 +1,21 @@
+# This CITATION.cff file was generated with cffinit.
+# Visit https://bit.ly/cffinit to generate yours today!
+
+cff-version: 1.2.0
+title: >-
+  MAM-E: Mammographic synthetic image generation with
+  diffusion models
+message: >-
+  If you use this software, please cite it using the
+  metadata from this file.
+type: software
+authors:
+  - given-names: Ricardo
+    family-names: Montoya-del-Angel
+    email: ricardo.montoya.da@gmail.com
+    affiliation: University of Girona
+  - given-names: Robert
+    family-names: Martí
+    email: robert.marti@udg.edu
+    affiliation: University of Girona
+repository-code: 'https://github.com/VICTORIA-project/mam-e'

envs/dreambooth_requirements.txt

@@ -29,7 +29,7 @@ dill==0.3.6
 docker-pycreds==0.4.0
 entrypoints==0.4
 executing==1.2.0
-fastapi==0.94.1
+fastapi==0.109.1
 ffmpy==0.3.0
 filelock==3.10.0
 flit_core @ file:///opt/conda/conda-bld/flit-core_1644941570762/work/source/flit_core
@@ -39,7 +39,7 @@ fsspec==2023.3.0
 ftfy==6.1.1
 gitdb==4.0.10
 GitPython==3.1.41
-gradio==3.34.0
+gradio==4.19.2
 h11==0.14.0
 httpcore==0.16.3
 httpx==0.23.3
@@ -78,7 +78,7 @@ ninja==1.11.1
 numpy @ file:///croot/numpy_and_numpy_base_1672336185480/work
 omidb==0.7.0
 opencv-python==4.7.0.72
-orjson==3.8.7
+orjson==3.9.15
 packaging==23.0
 pandas==1.5.3
 parso==0.8.3
@@ -105,7 +105,7 @@ pyre-extensions==0.0.23
 pyrsistent==0.19.3
 PySocks @ file:///tmp/build/80754af9/pysocks_1605305812635/work
 python-dateutil==2.8.2
-python-multipart==0.0.6
+python-multipart==0.0.7
 pytz==2022.7.1
 PyYAML==6.0
 pyzmq==25.0.1
@@ -121,7 +121,7 @@ six==1.16.0
 smmap==5.0.0
 sniffio==1.3.0
 stack-data==0.6.2
-starlette==0.26.1
+starlette==0.36.2
 subprocess32==3.5.4
 tokenizers==0.13.2
 toolz==0.12.0
@@ -132,7 +132,7 @@ torchvision==0.14.1
 tornado==6.2
 tqdm==4.65.0
 traitlets==5.9.0
-transformers==4.30.0
+transformers==4.36.0
 triton==2.0.0.post1
 typing-inspect==0.8.0
 typing_extensions @ file:///croot/typing_extensions_1669924550328/work

experiments/inference/fusion/mam-e_fusion.py

@@ -7,7 +7,7 @@
 sys.path.insert(0,str(repo_path)) if str(repo_path) not in sys.path else None
 
 os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"   # see issue #152
-os.environ["CUDA_VISIBLE_DEVICES"]="1"
+os.environ["CUDA_VISIBLE_DEVICES"]="0"
 
 from diffusers import DPMSolverMultistepScheduler, StableDiffusionPipeline
 import torch