Hide GPU

Dockerfile

@@ -19,7 +19,7 @@ ENV HF_HOME=/home/appuser/.cache/huggingface \
     TRANSFORMERS_CACHE=/home/appuser/.cache/huggingface \
     TORCH_HOME=/home/appuser/.cache/huggingface/torch \
     PATH="/home/appuser/.local/bin:${PATH}"
-
+ENV CUDA_VISIBLE_DEVICES=""
 # (the extra PATH entry suppresses the “not on PATH” pip warnings)
 
 # —— Python deps (NO diffvg here) ————————————————

code/losses.py

@@ -35,7 +35,7 @@ class SDSLoss(nn.Module):
         self.embed_text()
 
         # 2️⃣  NOW turn on off-loading (only UNet & VAE get meta tensors)
-        self.pipe.enable_model_cpu_offload()
+        #self.pipe.enable_model_cpu_offload()
 
         # text-encoder is no longer needed
         del self.pipe.text_encoder, self.pipe.tokenizer