harpomaxx commited on
Commit
e5e5614
·
verified ·
1 Parent(s): 2054eae

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -22
app.py CHANGED
@@ -121,26 +121,6 @@ with gr.Blocks() as demo:
121
 
122
  if __name__ == "__main__":
123
 
124
- parser = argparse.ArgumentParser()
125
-
126
- parser.add_argument(
127
- "--ip",
128
- default="0.0.0.0",
129
- help="The IP address to of the server"
130
- )
131
- parser.add_argument(
132
- "--port",
133
- type=int,
134
- default=7860,
135
- help="The port used"
136
- )
137
- parser.add_argument(
138
- "--gpuid",
139
- default="0",
140
- help="The gpu id"
141
- )
142
-
143
- args = parser.parse_args()
144
- pipe = StableDiffusionPipeline.from_single_file(PATH,torch_dtype=torch.float16).to(f"cuda:{args.gpuid}")
145
  demo.queue(concurrency_count=2,
146
- ).launch(server_name = args.ip, server_port = args.port)
 
121
 
122
  if __name__ == "__main__":
123
 
124
+ pipe = StableDiffusionPipeline.from_single_file(PATH,torch_dtype=torch.float16).to("cuda")
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
125
  demo.queue(concurrency_count=2,
126
+ ).launch()