encounter1997 commited on
Commit
b2d1afb
Β·
1 Parent(s): 5d906f6

update p2p model path

Browse files
vid2vid_zero/p2p/null_text_w_ptp.py CHANGED
@@ -32,7 +32,7 @@ MAX_NUM_WORDS = 77
32
  device = torch.device('cuda')
33
  from transformers import CLIPTextModel, CLIPTokenizer
34
 
35
- pretrained_model_path = "checkpoints/stable-diffusion-v1-4/"
36
 
37
  ldm_stable = None
38
  tokenizer = CLIPTokenizer.from_pretrained(pretrained_model_path, subfolder="tokenizer")
 
32
  device = torch.device('cuda')
33
  from transformers import CLIPTextModel, CLIPTokenizer
34
 
35
+ pretrained_model_path = "checkpoints/CompVis/stable-diffusion-v1-4/"
36
 
37
  ldm_stable = None
38
  tokenizer = CLIPTokenizer.from_pretrained(pretrained_model_path, subfolder="tokenizer")
vid2vid_zero/p2p/p2p_stable.py CHANGED
@@ -21,7 +21,7 @@ from . import ptp_utils
21
  from . import seq_aligner
22
  from transformers import CLIPTextModel, CLIPTokenizer
23
 
24
- pretrained_model_path = "checkpoints/stable-diffusion-v1-4/"
25
  ldm_stable = None
26
  tokenizer = CLIPTokenizer.from_pretrained(pretrained_model_path, subfolder="tokenizer")
27
 
 
21
  from . import seq_aligner
22
  from transformers import CLIPTextModel, CLIPTokenizer
23
 
24
+ pretrained_model_path = "checkpoints/CompVis/stable-diffusion-v1-4/"
25
  ldm_stable = None
26
  tokenizer = CLIPTokenizer.from_pretrained(pretrained_model_path, subfolder="tokenizer")
27