Trying fix variants
Browse files- app.py +5 -0
- 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
|