philipp-zettl commited on
Commit
c64ef31
1 Parent(s): 09bd06d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -4,10 +4,9 @@ import spaces
4
  import torch
5
  from diffusers import DiffusionPipeline
6
 
7
-
8
- pipe = DiffusionPipeline.from_pretrained(
9
  "philipp-zettl/jon_juarez-lora",
10
- torch_dtype=torch.float32,
11
  hf_token=os.environ.get('HF_TOKEN')
12
  )
13
  pipe.to('cuda')
 
4
  import torch
5
  from diffusers import DiffusionPipeline
6
 
7
+ pipeline = DiffusionPipeline.from_pretrained("segmind/tiny-sd")
8
+ pipeline.load_lora_weights(
9
  "philipp-zettl/jon_juarez-lora",
 
10
  hf_token=os.environ.get('HF_TOKEN')
11
  )
12
  pipe.to('cuda')