Simple command line tool for text to image generation using OpenAI's CLIP and Siren (Implicit neural representation network). Technique was originally created by https://twitter.com/advadnoun
Traceback (most recent call last): File "c:\users\adria\appdata\local\programs\python\python39\lib\runpy.py", line 197, in _run_module_as_main return _run_code(code, main_globals, None, File "c:\users\adria\appdata\local\programs\python\python39\lib\runpy.py", line 87, in _run_code exec(code, run_globals) File "C:\Users\adria\AppData\Local\Programs\Python\Python39\Scripts\imagine.exe\__main__.py", line 7, in <module> File "c:\users\adria\appdata\local\programs\python\python39\lib\site-packages\deep_daze\cli.py", line 111, in main fire.Fire(train) File "c:\users\adria\appdata\local\programs\python\python39\lib\site-packages\fire\core.py", line 141, in Fire component_trace = _Fire(component, args, parsed_flag_args, context, name) File "c:\users\adria\appdata\local\programs\python\python39\lib\site-packages\fire\core.py", line 466, in _Fire component, remaining_args = _CallAndUpdateTrace( File "c:\users\adria\appdata\local\programs\python\python39\lib\site-packages\fire\core.py", line 681, in _CallAndUpdateTrace component = fn(*varargs, **kwargs) File "c:\users\adria\appdata\local\programs\python\python39\lib\site-packages\deep_daze\cli.py", line 107, in train imagine() File "c:\users\adria\appdata\local\programs\python\python39\lib\site-packages\torch\nn\modules\module.py", line 889, in _call_impl result = self.forward(*input, **kwargs) File "c:\users\adria\appdata\local\programs\python\python39\lib\site-packages\deep_daze\deep_daze.py", line 447, in forward _, loss = self.train_step(epoch, i) File "c:\users\adria\appdata\local\programs\python\python39\lib\site-packages\deep_daze\deep_daze.py", line 380, in train_step out, loss = self.model(self.clip_encoding) File "c:\users\adria\appdata\local\programs\python\python39\lib\site-packages\torch\nn\modules\module.py", line 889, in _call_impl result = self.forward(*input, **kwargs) File "c:\users\adria\appdata\local\programs\python\python39\lib\site-packages\deep_daze\deep_daze.py", line 168, in forward out = self.model() File "c:\users\adria\appdata\local\programs\python\python39\lib\site-packages\torch\nn\modules\module.py", line 889, in _call_impl result = self.forward(*input, **kwargs) File "c:\users\adria\appdata\local\programs\python\python39\lib\site-packages\siren_pytorch\siren_pytorch.py", line 97, in forward out = self.net(coords) File "c:\users\adria\appdata\local\programs\python\python39\lib\site-packages\torch\nn\modules\module.py", line 889, in _call_impl result = self.forward(*input, **kwargs) File "c:\users\adria\appdata\local\programs\python\python39\lib\site-packages\siren_pytorch\siren_pytorch.py", line 76, in forward x = self.net(x) File "c:\users\adria\appdata\local\programs\python\python39\lib\site-packages\torch\nn\modules\module.py", line 889, in _call_impl result = self.forward(*input, **kwargs) File "c:\users\adria\appdata\local\programs\python\python39\lib\site-packages\torch\nn\modules\container.py", line 119, in forward input = module(input) File "c:\users\adria\appdata\local\programs\python\python39\lib\site-packages\torch\nn\modules\module.py", line 889, in _call_impl result = self.forward(*input, **kwargs) File "c:\users\adria\appdata\local\programs\python\python39\lib\site-packages\siren_pytorch\siren_pytorch.py", line 48, in forward out = self.activation(out) File "c:\users\adria\appdata\local\programs\python\python39\lib\site-packages\torch\nn\modules\module.py", line 889, in _call_impl result = self.forward(*input, **kwargs) File "c:\users\adria\appdata\local\programs\python\python39\lib\site-packages\siren_pytorch\siren_pytorch.py", line 19, in forward return torch.sin(self.w0 * x) RuntimeError: CUDA out of memory. Tried to allocate 128.00 MiB (GPU 0; 8.00 GiB total capacity; 1.84 GiB already allocated; 119.94 MiB free; 1.86 GiB reserved in total by PyTorch)
Lucagamer22006
commented
3 years ago
Traceback (most recent call last): File "c:\users\adria\appdata\local\programs\python\python39\lib\runpy.py", line 197, in _run_module_as_main return _run_code(code, main_globals, None, File "c:\users\adria\appdata\local\programs\python\python39\lib\runpy.py", line 87, in _run_code exec(code, run_globals) File "C:\Users\adria\AppData\Local\Programs\Python\Python39\Scripts\imagine.exe\__main__.py", line 7, in <module> File "c:\users\adria\appdata\local\programs\python\python39\lib\site-packages\deep_daze\cli.py", line 111, in main fire.Fire(train) File "c:\users\adria\appdata\local\programs\python\python39\lib\site-packages\fire\core.py", line 141, in Fire component_trace = _Fire(component, args, parsed_flag_args, context, name) File "c:\users\adria\appdata\local\programs\python\python39\lib\site-packages\fire\core.py", line 466, in _Fire component, remaining_args = _CallAndUpdateTrace( File "c:\users\adria\appdata\local\programs\python\python39\lib\site-packages\fire\core.py", line 681, in _CallAndUpdateTrace component = fn(*varargs, **kwargs) File "c:\users\adria\appdata\local\programs\python\python39\lib\site-packages\deep_daze\cli.py", line 107, in train imagine() File "c:\users\adria\appdata\local\programs\python\python39\lib\site-packages\torch\nn\modules\module.py", line 889, in _call_impl result = self.forward(*input, **kwargs) File "c:\users\adria\appdata\local\programs\python\python39\lib\site-packages\deep_daze\deep_daze.py", line 447, in forward _, loss = self.train_step(epoch, i) File "c:\users\adria\appdata\local\programs\python\python39\lib\site-packages\deep_daze\deep_daze.py", line 380, in train_step out, loss = self.model(self.clip_encoding) File "c:\users\adria\appdata\local\programs\python\python39\lib\site-packages\torch\nn\modules\module.py", line 889, in _call_impl result = self.forward(*input, **kwargs) File "c:\users\adria\appdata\local\programs\python\python39\lib\site-packages\deep_daze\deep_daze.py", line 168, in forward out = self.model() File "c:\users\adria\appdata\local\programs\python\python39\lib\site-packages\torch\nn\modules\module.py", line 889, in _call_impl result = self.forward(*input, **kwargs) File "c:\users\adria\appdata\local\programs\python\python39\lib\site-packages\siren_pytorch\siren_pytorch.py", line 97, in forward out = self.net(coords) File "c:\users\adria\appdata\local\programs\python\python39\lib\site-packages\torch\nn\modules\module.py", line 889, in _call_impl result = self.forward(*input, **kwargs) File "c:\users\adria\appdata\local\programs\python\python39\lib\site-packages\siren_pytorch\siren_pytorch.py", line 76, in forward x = self.net(x) File "c:\users\adria\appdata\local\programs\python\python39\lib\site-packages\torch\nn\modules\module.py", line 889, in _call_impl result = self.forward(*input, **kwargs) File "c:\users\adria\appdata\local\programs\python\python39\lib\site-packages\torch\nn\modules\container.py", line 119, in forward input = module(input) File "c:\users\adria\appdata\local\programs\python\python39\lib\site-packages\torch\nn\modules\module.py", line 889, in _call_impl result = self.forward(*input, **kwargs) File "c:\users\adria\appdata\local\programs\python\python39\lib\site-packages\siren_pytorch\siren_pytorch.py", line 48, in forward out = self.activation(out) File "c:\users\adria\appdata\local\programs\python\python39\lib\site-packages\torch\nn\modules\module.py", line 889, in _call_impl result = self.forward(*input, **kwargs) File "c:\users\adria\appdata\local\programs\python\python39\lib\site-packages\siren_pytorch\siren_pytorch.py", line 19, in forward return torch.sin(self.w0 * x) RuntimeError: CUDA out of memory. Tried to allocate 128.00 MiB (GPU 0; 8.00 GiB total capacity; 1.84 GiB already allocated; 119.94 MiB free; 1.86 GiB reserved in total by PyTorch)