dezzman commited on
Commit
7e0751a
·
verified ·
1 Parent(s): df4b467

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -36,6 +36,9 @@ def infer(
36
  ):
37
  generator = torch.Generator().manual_seed(seed)
38
 
 
 
 
39
  if model_id != model_repo_id_default:
40
  pipe = DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch_dtype)
41
  pipe = pipe.to(device)
 
36
  ):
37
  generator = torch.Generator().manual_seed(seed)
38
 
39
+ print('!!!!!!')
40
+ print(model_id)
41
+ print('!!!!!!')
42
  if model_id != model_repo_id_default:
43
  pipe = DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch_dtype)
44
  pipe = pipe.to(device)