trashchenkov commited on
Commit
606bc33
·
verified ·
1 Parent(s): 5d16941

Update app.py

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