It automatically stops when running in Colab #4
Description
Loading config from json: configs/default_config.json
2025-02-24 16:28:04.130575: E external/local_xla/xla/stream_executor/cuda/cuda_fft.cc:477] Unable to register cuFFT factory: Attempting to register factory for plugin cuFFT when one has already been registered
WARNING: All log messages before absl::InitializeLog() is called are written to STDERR
E0000 00:00:1740414484.530120 8769 cuda_dnn.cc:8310] Unable to register cuDNN factory: Attempting to register factory for plugin cuDNN when one has already been registered
E0000 00:00:1740414484.643028 8769 cuda_blas.cc:1418] Unable to register cuBLAS factory: Attempting to register factory for plugin cuBLAS when one has already been registered
2025-02-24 16:28:05.464498: I tensorflow/core/platform/cpu_feature_guard.cc:210] This TensorFlow binary is optimized to use available CPU instructions in performance-critical operations.
To enable the following instructions: AVX2 FMA, in other operations, rebuild TensorFlow with the appropriate compiler flags.
logging improved.
Number of GPUs: 0
Batch Size: 3
Max Epochs: 10
Dataset size: 1000
ControlLDM: Running in eps-prediction mode
DiffusionWrapper has 865.91 M params.
making attention of type 'vanilla-xformers' with 512 in_channels
building MemoryEfficientAttnBlock with 512 in_channels...
Working with z of shape (1, 4, 32, 32) = 4096 dimensions.
making attention of type 'vanilla-xformers' with 512 in_channels
building MemoryEfficientAttnBlock with 512 in_channels...
open_clip_model.safetensors: 100% 3.94G/3.94G [03:34<00:00, 18.4MB/s]
Loaded model config from [./models/cldm_v21.yaml]
Downloading pretrained model...
v2-1_512-ema-pruned.ckpt: 5.21GB [03:21, 25.9MB/s]
^C