Hugging Face's logo

Spaces:
mohansathya
/
twosd
Paused

App Files Community
mohansathya commited on
Commit
5211fb7
·
1 Parent(s): bb2badd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ from gradio.components import Textbox, Image
13
 
14
 
15
  repo_name = 'mohansathya/twosd' # YOUR REPO NAME
16
- pipe2 = StableDiffusionPipeline.from_pretrained(repo_name, torch_dtype=torch.bfloat16).to('cpu')
17
 
18
 
19
 
 
13
 
14
 
15
  repo_name = 'mohansathya/twosd' # YOUR REPO NAME
16
+ pipe2 = StableDiffusionPipeline.from_pretrained(repo_name, torch_dtype=torch.bfloat16)
17
 
18
 
19