ds1david commited on
Commit
eb1615e
Β·
1 Parent(s): f3b823f

Trying fix variants

Browse files
Files changed (2) hide show
  1. app.py +5 -0
  2. requirements.txt +0 -1
app.py CHANGED
@@ -21,6 +21,11 @@ pipe.load_lora_weights(
21
  peft_backend="peft" # This is crucial
22
  )
23
 
 
 
 
 
 
24
  print("Loading DPT Depth Model...")
25
  feature_extractor = DPTFeatureExtractor.from_pretrained("Intel/dpt-large")
26
  depth_model = DPTForDepthEstimation.from_pretrained("Intel/dpt-large").to(device)
 
21
  peft_backend="peft" # This is crucial
22
  )
23
 
24
+ # Converter os mΓ³dulos para float32 para evitar conflitos de dtype
25
+ pipe.unet = pipe.unet.float()
26
+ pipe.text_encoder = pipe.text_encoder.float()
27
+
28
+
29
  print("Loading DPT Depth Model...")
30
  feature_extractor = DPTFeatureExtractor.from_pretrained("Intel/dpt-large")
31
  depth_model = DPTForDepthEstimation.from_pretrained("Intel/dpt-large").to(device)
requirements.txt CHANGED
@@ -1,4 +1,3 @@
1
- open3d
2
  peft
3
  accelerate
4
  diffusers>=0.20.0
 
 
1
  peft
2
  accelerate
3
  diffusers>=0.20.0