jbilcke-hf HF Staff commited on
Commit
e7442ec
·
1 Parent(s): 75b3cc4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -36,7 +36,7 @@ if torch.cuda.is_available():
36
  variant='fp16')
37
 
38
  # LoRA for 360° images
39
- pipe.unet.load_lora_weights('artificialguybr/360Redmond', weight_name='View360.safetensors')
40
 
41
  if ENABLE_CPU_OFFLOAD:
42
  pipe.enable_model_cpu_offload()
 
36
  variant='fp16')
37
 
38
  # LoRA for 360° images
39
+ pipe.load_lora_weights('artificialguybr/360Redmond', weight_name='View360.safetensors')
40
 
41
  if ENABLE_CPU_OFFLOAD:
42
  pipe.enable_model_cpu_offload()