Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -3,7 +3,7 @@ import random
|
|
3 |
import gradio as gr
|
4 |
import numpy as np
|
5 |
|
6 |
-
import spaces
|
7 |
import torch
|
8 |
|
9 |
from inference_t2mv_sdxl import prepare_pipeline, run_pipeline
|
@@ -13,7 +13,8 @@ import transformers
|
|
13 |
transformers.utils.move_cache()
|
14 |
|
15 |
# Base model
|
16 |
-
base_model = "stabilityai/stable-diffusion-xl-base-1.0"
|
|
|
17 |
|
18 |
# Device and dtype
|
19 |
dtype = torch.bfloat16
|
@@ -38,7 +39,7 @@ pipe = prepare_pipeline(
|
|
38 |
)
|
39 |
|
40 |
|
41 |
-
|
42 |
def infer(
|
43 |
prompt,
|
44 |
seed=42,
|
@@ -187,4 +188,4 @@ Also try our other demos: [Text-to-Multiview (General)](https://huggingface.co/s
|
|
187 |
outputs=[result, seed],
|
188 |
)
|
189 |
|
190 |
-
demo.launch()
|
|
|
3 |
import gradio as gr
|
4 |
import numpy as np
|
5 |
|
6 |
+
#import spaces
|
7 |
import torch
|
8 |
|
9 |
from inference_t2mv_sdxl import prepare_pipeline, run_pipeline
|
|
|
13 |
transformers.utils.move_cache()
|
14 |
|
15 |
# Base model
|
16 |
+
#base_model = "stabilityai/stable-diffusion-xl-base-1.0"
|
17 |
+
base_model = "svjack/GenshinImpact_XL_Base"
|
18 |
|
19 |
# Device and dtype
|
20 |
dtype = torch.bfloat16
|
|
|
39 |
)
|
40 |
|
41 |
|
42 |
+
#@spaces.GPU()
|
43 |
def infer(
|
44 |
prompt,
|
45 |
seed=42,
|
|
|
188 |
outputs=[result, seed],
|
189 |
)
|
190 |
|
191 |
+
demo.launch(share = True)
|