Fixed requierements

patch/python.py

@@ -22,10 +22,10 @@ from huggingface_hub import hf_hub_download
 token =  os.getenv("HF_TKN")
 # Merge both mappings
 NODE_CLASS_MAPPINGS = {**NODE_CLASS_MAPPINGS_1, **NODE_CLASS_MAPPINGS_2, **NODE_CLASS_MAPPINGS_3, **NODE_CLASS_MAPPINGS_4, **NODE_CLASS_MAPPINGS_5, **NODE_CLASS_MAPPINGS_6}
-hf_hub_download(repo_id="black-forest-labs/FLUX.1-dev", filename="flux1-dev.safetensors", local_dir="models/unet", token = token)
-hf_hub_download(repo_id="black-forest-labs/FLUX.1-dev", filename="ae.safetensors", local_dir="models/vae", token = token)
-hf_hub_download(repo_id="comfyanonymous/flux_text_encoders", filename="clip_l.safetensors", local_dir="models/text_encoders", token = token)
-hf_hub_download(repo_id="comfyanonymous/flux_text_encoders", filename="t5xxl_fp16.safetensors", local_dir="models/text_encoders", token = token)
+#hf_hub_download(repo_id="black-forest-labs/FLUX.1-dev", filename="flux1-dev.safetensors", local_dir="models/unet", token = token)
+#hf_hub_download(repo_id="black-forest-labs/FLUX.1-dev", filename="ae.safetensors", local_dir="models/vae", token = token)
+#hf_hub_download(repo_id="comfyanonymous/flux_text_encoders", filename="clip_l.safetensors", local_dir="models/text_encoders", token = token)
+#hf_hub_download(repo_id="comfyanonymous/flux_text_encoders", filename="t5xxl_fp16.safetensors", local_dir="models/text_encoders", token = token)
 
 def preprocess_image_tensor(image):
     # If image has a batch dimension (shape: [1, C, H, W]), remove it.

requirements.txt

@@ -2,28 +2,28 @@
 torch==2.7.0
 torchvision==0.22.0
 torchaudio==2.7.0
-torchsde
-numpy>=1.25.0
-einops
-transformers>=4.28.1
-tokenizers>=0.13.3
-sentencepiece
-safetensors>=0.4.2
-aiohttp>=3.11.8
-yarl>=1.18.0
-pyyaml
-Pillow
-scipy
-tqdm
-psutil
-kornia>=0.7.1
-spandrel
-soundfile
-av
+torchsde==0.2.6
+numpy==1.26.4
+einops== 0.8.1
+transformers==4.52.4
+tokenizers==0.21.1
+sentencepiece==0.2.0
+safetensors==0.5.3
+aiohtt==3.12.9
+yarl==1.20.0
+pyyaml==6.0.2
+Pillow==10.4.0
+scipy==1.15.3
+tqdm==4.67.1
+psutil=7.0.0
+kornia==0.8.1
+spandrel==0.4.1
+soundfile==0.13.1
+av==14.4.0
 accelerate
 diffusers==0.33.1
-xformer
-sentencepiece
+xformer==1.7.0
+sentencepiece==0.2.0
 peft==0.15.0
 gradio==4.44.1
 pydantic==2.10.6

start_docker.sh

@@ -1,8 +1,11 @@
 #!/bin/bash
 # Start docker container in WSL to debug locally
 NAME=T2A
+WEIGHTS=/mnt/e/weights
+
 IMAGE_NAME=text2autochrome
-docker run --gpus all -it -d --rm \
+docker run --gpus all -it --rm \
   --name=$NAME \
+  -v $WEIGHTS:/workspace/autochrome-gen-cui/models \
   $IMAGE_NAME