svjack commited on
Commit
1834741
·
verified ·
1 Parent(s): 47795ab

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
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
- @spaces.GPU()
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)