Initial commit

src/utils/gradio_utils.py

@@ -208,8 +208,8 @@ def launch_main(dest, step, fpath_z_gen, seed):
     pretrained_model_name = 'CompVis/stable-diffusion-v1-4'   
     scheduler = DDIMScheduler(beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear", clip_sample=False, set_alpha_to_one=False) 
     pipe = StableDiffusionPipeline.from_pretrained(pretrained_model_name, scheduler=scheduler, torch_dtype=torch.float16).to("cuda")
-    tokenizer = CLIPTokenizer.from_pretrained(pretrained_model_name, subfolder="tokenizer", use_auth_token=True)
-    CLIP_text_encoder = CLIPTextModel.from_pretrained(pretrained_model_name, subfolder="text_encoder", use_auth_token=True)
+    tokenizer = CLIPTokenizer.from_pretrained(pretrained_model_name, subfolder="tokenizer")#, use_auth_token=True)
+    CLIP_text_encoder = CLIPTextModel.from_pretrained(pretrained_model_name, subfolder="text_encoder")#, use_auth_token=True)
 
 
     # Encode the target text.