Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -34,14 +34,7 @@ from torchvision import transforms
|
|
| 34 |
from models.controlnet import ControlNetModel
|
| 35 |
from models.unet_2d_condition import UNet2DConditionModel
|
| 36 |
|
| 37 |
-
|
| 38 |
-
|
| 39 |
-
# vlm_model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
|
| 40 |
-
# VLM_NAME,
|
| 41 |
-
# torch_dtype="auto",
|
| 42 |
-
# device_map="auto" # immediately dispatches layers onto available GPUs
|
| 43 |
-
# )
|
| 44 |
-
# vlm_processor = AutoProcessor.from_pretrained(VLM_NAME)
|
| 45 |
|
| 46 |
def _generate_vlm_prompt(
|
| 47 |
vlm_model: Qwen2_5_VLForConditionalGeneration,
|
|
|
|
| 34 |
from models.controlnet import ControlNetModel
|
| 35 |
from models.unet_2d_condition import UNet2DConditionModel
|
| 36 |
|
| 37 |
+
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 38 |
|
| 39 |
def _generate_vlm_prompt(
|
| 40 |
vlm_model: Qwen2_5_VLForConditionalGeneration,
|