JackAILab commited on
Commit
e346b2b
·
verified ·
1 Parent(s): 176d22b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -2,13 +2,14 @@ import gradio as gr
2
  import torch
3
  import os
4
  import glob
 
 
5
  from datetime import datetime
6
  from PIL import Image
7
  from diffusers.utils import load_image
8
  from diffusers import EulerDiscreteScheduler
9
  from pipline_StableDiffusion_ConsistentID import ConsistentIDStableDiffusionPipeline\
10
  from huggingface_hub import hf_hub_download
11
- import spaces
12
 
13
  zero = torch.Tensor([0]).cuda()
14
  print(zero.device) # <-- 'cpu' 🤔
 
2
  import torch
3
  import os
4
  import glob
5
+ import spaces
6
+
7
  from datetime import datetime
8
  from PIL import Image
9
  from diffusers.utils import load_image
10
  from diffusers import EulerDiscreteScheduler
11
  from pipline_StableDiffusion_ConsistentID import ConsistentIDStableDiffusionPipeline\
12
  from huggingface_hub import hf_hub_download
 
13
 
14
  zero = torch.Tensor([0]).cuda()
15
  print(zero.device) # <-- 'cpu' 🤔