Kunhak Lee commited on
Commit
4ddb8f6
1 Parent(s): 347fd66

Add multi_view_diffusion_pipeline and some import

Browse files
Files changed (1) hide show
  1. app.py +10 -0
app.py CHANGED
@@ -1,4 +1,14 @@
1
  import gradio as gr
 
 
 
 
 
 
 
 
 
 
2
 
3
  def run(image):
4
  image = np.array(image, dtype=np.float32) / 255.0
 
1
  import gradio as gr
2
+ import torch
3
+ from diffusers import DiffusionPipeline
4
+ import numpy as np
5
+
6
+ multi_view_diffusion_pipeline = DiffusionPipeline.from_pretrained(
7
+ "2gnak/multi-view-diffusion-demo",
8
+ custom_pipeline="dylanebert/multi-view-diffusion",
9
+ torch_dtype=torch.float16,
10
+ trust_remote_code=True,
11
+ ).to("cuda")
12
 
13
  def run(image):
14
  image = np.array(image, dtype=np.float32) / 255.0