lemonaddie commited on
Commit
366fb40
1 Parent(s): c1e3a9e

Update app_recon.py

Browse files
Files changed (1) hide show
  1. app_recon.py +0 -7
app_recon.py CHANGED
@@ -55,13 +55,6 @@ from torchvision.transforms import InterpolationMode
55
 
56
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
57
 
58
- # stable_diffusion_repo_path = "stabilityai/stable-diffusion-2-1-unclip"
59
- # sd_image_variations_diffusers_path = 'lambdalabs/sd-image-variations-diffusers'
60
- # vae = AutoencoderKL.from_pretrained(stable_diffusion_repo_path, subfolder='vae')
61
- # scheduler = DDIMScheduler.from_pretrained(stable_diffusion_repo_path, subfolder='scheduler')
62
- # image_encoder = CLIPVisionModelWithProjection.from_pretrained(sd_image_variations_diffusers_path, subfolder="image_encoder")
63
- # feature_extractor = CLIPImageProcessor.from_pretrained(sd_image_variations_diffusers_path, subfolder="feature_extractor")
64
-
65
  vae = AutoencoderKL.from_pretrained("./", subfolder='vae')
66
  scheduler = DDIMScheduler.from_pretrained("./", subfolder='scheduler')
67
  image_encoder = CLIPVisionModelWithProjection.from_pretrained("./", subfolder="image_encoder")
 
55
 
56
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
57
 
 
 
 
 
 
 
 
58
  vae = AutoencoderKL.from_pretrained("./", subfolder='vae')
59
  scheduler = DDIMScheduler.from_pretrained("./", subfolder='scheduler')
60
  image_encoder = CLIPVisionModelWithProjection.from_pretrained("./", subfolder="image_encoder")