lucidrains / big-sleep

A simple command line tool for text to image generation, using OpenAI's CLIP and a BigGAN. Technique was originally created by https://twitter.com/advadnoun
MIT License
2.57k stars 304 forks source link

Checkpoint #128

Open dasbts opened 2 years ago

dasbts commented 2 years ago

Hello,

Is it possible to add a checkpoint to continue running? I'm currently somewhere where Colab Pro is not possible, and I want to continue iterating.

I looked into the files and it seems like the OpenAI CLIP contains: def load(name: str, device: Union[str, torch.device] = "cuda" if torch.cuda.is_available() else "cpu", jit=True): """Load a CLIP model

Parameters
----------
name : str
    A model name listed by `clip.available_models()`, or the path to a model checkpoint containing the state_dict 

If it's possible to make checkpoints, would this be a good place to do so?

Thanks in advance, amazing work on this Github.