Hugging Face's logo

Spaces:
Nymbo
/
turbo_edit
Paused

App Files Community
turboedit commited on
Commit
128983c
•
1 Parent(s): 866e10c

Update my_run.py

Browse files
Files changed (1) hide show
  1. my_run.py +1 -1
my_run.py CHANGED
@@ -8,7 +8,7 @@ from PIL import Image
8
  num_steps_inversion = 5
9
  strngth = 0.8
10
  generator = None
11
- device = "cuda"
12
  image_path = "edit_dataset/01.jpg"
13
  src_prompt = "butterfly perched on purple flower"
14
  tgt_prompt = "dragonfly perched on purple flower"
 
8
  num_steps_inversion = 5
9
  strngth = 0.8
10
  generator = None
11
+ device = "cuda" if torch.cuda.is_available() else "cpu"
12
  image_path = "edit_dataset/01.jpg"
13
  src_prompt = "butterfly perched on purple flower"
14
  tgt_prompt = "dragonfly perched on purple flower"