fffiloni commited on
Commit
51b7f4f
1 Parent(s): 34aa211

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -12
app.py CHANGED
@@ -41,21 +41,21 @@ def train(prompt, init_image, trn_text, trn_steps):
41
  return "Training is finished !"
42
 
43
  def generate(prompt, init_image, trn_text, trn_steps):
44
- #init_image = Image.open(init_image).convert("RGB")
45
- #init_image = init_image.resize((256, 256))
46
 
47
 
48
- #res = pipe.train(
49
- # prompt,
50
- # init_image,
51
- # guidance_scale=7.5,
52
- # num_inference_steps=50,
53
- # generator=generator,
54
- # text_embedding_optimization_steps=trn_text,
55
- # model_fine_tuning_optimization_steps=trn_steps)
56
 
57
- #with torch.no_grad():
58
- # torch.cuda.empty_cache()
59
 
60
 
61
 
 
41
  return "Training is finished !"
42
 
43
  def generate(prompt, init_image, trn_text, trn_steps):
44
+ init_image = Image.open(init_image).convert("RGB")
45
+ init_image = init_image.resize((256, 256))
46
 
47
 
48
+ res = pipe.train(
49
+ prompt,
50
+ init_image,
51
+ guidance_scale=7.5,
52
+ num_inference_steps=50,
53
+ generator=generator,
54
+ text_embedding_optimization_steps=trn_text,
55
+ model_fine_tuning_optimization_steps=trn_steps)
56
 
57
+ with torch.no_grad():
58
+ torch.cuda.empty_cache()
59
 
60
 
61