Update app.py
Browse files
app.py
CHANGED
@@ -121,26 +121,6 @@ with gr.Blocks() as demo:
|
|
121 |
|
122 |
if __name__ == "__main__":
|
123 |
|
124 |
-
|
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(
|
|
|
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()
|