Hugging Face's logo

Spaces:
Clifx
/
PS2.ai
Running

App Files Community
Linoy Tsaban commited on
Commit
27e096e
•
1 Parent(s): e79152d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -49,7 +49,7 @@ sd_model_id = "runwayml/stable-diffusion-v1-5"
49
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
50
  sd_pipe = StableDiffusionPipeline.from_pretrained(sd_model_id).to(device)
51
  sd_pipe.scheduler = DDIMScheduler.from_config(sd_model_id, subfolder = "scheduler")
52
- sem_pipe = SemanticStableDiffusionPipeline.from_pretrained(sd_model_id, torch_dtype=torch.float16).to(device)
53
 
54
 
55
  def edit(input_image, input_image_prompt='', target_prompt='', edit_prompt='',
 
49
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
50
  sd_pipe = StableDiffusionPipeline.from_pretrained(sd_model_id).to(device)
51
  sd_pipe.scheduler = DDIMScheduler.from_config(sd_model_id, subfolder = "scheduler")
52
+ sem_pipe = SemanticStableDiffusionPipeline.from_pretrained(sd_model_id).to(device)
53
 
54
 
55
  def edit(input_image, input_image_prompt='', target_prompt='', edit_prompt='',