treasuraid commited on
Commit
71ed815
·
1 Parent(s): 3ff99e6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -8
app.py CHANGED
@@ -17,17 +17,10 @@ SAMPLE_IMAGE = 'weights/Sample.png'
17
 
18
  def load_pipeline_w_lora() :
19
 
20
- # Load pretrained unet from huggingface
21
- unet = UNet2DConditionModel.from_pretrained(
22
- MODEL_REPO,
23
- subfolder="unet",
24
- revision=None
25
- )
26
-
27
  # Load pipeline
28
  pipeline = DiffusionPipeline.from_pretrained(
29
  MODEL_REPO,
30
- unet=unet,
31
  revision=None,
32
  torch_dtype=torch.float32,
33
  )
 
17
 
18
  def load_pipeline_w_lora() :
19
 
20
+
 
 
 
 
 
 
21
  # Load pipeline
22
  pipeline = DiffusionPipeline.from_pretrained(
23
  MODEL_REPO,
 
24
  revision=None,
25
  torch_dtype=torch.float32,
26
  )