Futuretop commited on
Commit
99287f4
·
verified ·
1 Parent(s): c740e13

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -2,6 +2,8 @@ import gradio as gr
2
  import torch
3
  import random
4
  from PIL import Image
 
 
5
  from transformers import AutoProcessor, AutoModelForCausalLM
6
  from diffusers import DiffusionPipeline, FlowMatchEulerDiscreteScheduler
7
 
 
2
  import torch
3
  import random
4
  from PIL import Image
5
+ import modeling_florence2
6
+ modeling_florence2.DaViT._initialize_weights = lambda self, module=None: None
7
  from transformers import AutoProcessor, AutoModelForCausalLM
8
  from diffusers import DiffusionPipeline, FlowMatchEulerDiscreteScheduler
9