Linoy Tsaban commited on
Commit
a00cde7
1 Parent(s): 066c23c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -45,7 +45,7 @@ def sample(wt, zs, wts, prompt_tar="", cfg_scale_tar=15, skip=36, eta = 1):
45
  # load pipelines
46
  sd_model_id = "runwayml/stable-diffusion-v1-5"
47
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
48
- sd_pipe = StableDiffusionPipeline.from_pretrained(sd_model_id, ,torch_dtype=torch.float16).to(device)
49
  sd_pipe.scheduler = DDIMScheduler.from_config(sd_model_id, subfolder = "scheduler")
50
  sem_pipe = SemanticStableDiffusionPipeline.from_pretrained(sd_model_id, ,torch_dtype=torch.float16).to(device)
51
 
 
45
  # load pipelines
46
  sd_model_id = "runwayml/stable-diffusion-v1-5"
47
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
48
+ sd_pipe = StableDiffusionPipeline.from_pretrained(sd_model_id, torch_dtype=torch.float16).to(device)
49
  sd_pipe.scheduler = DDIMScheduler.from_config(sd_model_id, subfolder = "scheduler")
50
  sem_pipe = SemanticStableDiffusionPipeline.from_pretrained(sd_model_id, ,torch_dtype=torch.float16).to(device)
51