Changes to make this install and run cleanly on Windows.

README.md

@@ -20,7 +20,7 @@
 
 ```
 git clone https://github.com/omerbt/Text2LIVE.git
-conda create --name text2live python=3.9 
+conda create --name text2live python=3.9 pytorch==1.10.0 torchvision==0.11.0 torchaudio==0.10.0 cudatoolkit=11.3 -c pytorch -c conda-forge
 conda activate text2live 
 pip install -r requirements.txt
 ```

requirements.txt

@@ -1,6 +1,4 @@
 pillow
-torch~=1.10.0
-torchvision~=0.11.2
 tqdm
 numpy
 ftfy

train_image.py

@@ -21,7 +21,7 @@ def train_model(config):
     # set seed
     seed = config["seed"]
     if seed == -1:
-        seed = np.random.randint(2 ** 32)
+        seed = np.random.randint(2 ** 32, dtype=np.int64)
         random.seed(seed)
         np.random.seed(seed)
         torch.manual_seed(seed)
@@ -128,4 +128,4 @@ def save_locally(results_folder, log_data):
 
     train_model(config)
     if config["use_wandb"]:
-        wandb.finish()
+        wandb.finish()

train_video.py

@@ -19,7 +19,7 @@ def train_model(config):
     # set seed
     seed = config["seed"]
     if seed == -1:
-        seed = np.random.randint(2 ** 32)
+        seed = np.random.randint(2 ** 32, dtype=np.int64)
     random.seed(seed)
     np.random.seed(seed)
     torch.manual_seed(seed)