Merge remote-tracking branch 'upstream/main' into main

README.md

@@ -42,6 +42,10 @@
 
 *the tragic intimacy of the eternal conversation with oneself* - by <a href="https://github.com/moirage">moirage</a>
 
+<img src="./samples/demon_fire.png" width="250px"></img>
+
+*demon fire* - by <a href="https://github.com/WiseNat">WiseNat</a>
+
 ## Big Sleep
 
 <a href="https://twitter.com/advadnoun">Ryan Murdock</a> has done it again, combining OpenAI's <a href="https://github.com/openai/CLIP">CLIP</a> and the generator from a <a href="https://arxiv.org/abs/1809.11096">BigGAN</a>! This repository wraps up his work so it is easily accessible to anyone who owns a GPU.

big_sleep/big_sleep.py

@@ -2,6 +2,9 @@
 import sys
 import subprocess
 import signal
+import string
+import re
+
 from datetime import datetime
 from pathlib import Path
 import random
@@ -18,7 +21,7 @@
 from big_sleep.ema import EMA
 from big_sleep.resample import resample
 from big_sleep.biggan import BigGAN
-from big_sleep.clip import load, tokenize, normalize_image
+from big_sleep.clip import load, tokenize
 
 assert torch.cuda.is_available(), 'CUDA must be available in order to use Big Sleep'
 
@@ -129,7 +132,7 @@ def rand_cutout(image, size, center_bias=False, center_focus=2):
 
 # load clip
 
-perceptor, preprocess = load()
+perceptor, normalize_image = load('ViT-B/32', jit = False)
 
 # load biggan