Skip to content

Commit

Permalink
Fix gmflow_dir (#6583)
Browse files Browse the repository at this point in the history
* remove sys.path

* update readme
  • Loading branch information
pravdomil authored Mar 9, 2024
1 parent ec64f37 commit e64fdcf
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 12 deletions.
8 changes: 3 additions & 5 deletions examples/community/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -3414,15 +3414,13 @@ pipeline(prompt, uncond, inverted_latent, guidance_scale=7.5, num_inference_step

### Rerender A Video

This is the Diffusers implementation of zero-shot video-to-video translation pipeline [Rerender A Video](https://github.com/williamyang1991/Rerender_A_Video) (without Ebsynth postprocessing). To run the code, please install gmflow. Then modify the path in `examples/community/rerender_a_video.py`:
This is the Diffusers implementation of zero-shot video-to-video translation pipeline [Rerender A Video](https://github.com/williamyang1991/Rerender_A_Video) (without Ebsynth postprocessing). To run the code, please install gmflow. Then modify the path in `gmflow_dir`. After that, you can run the pipeline with:

```py
import sys
gmflow_dir = "/path/to/gmflow"
```

After that, you can run the pipeline with:
sys.path.insert(0, gmflow_dir)

```py
from diffusers import ControlNetModel, AutoencoderKL, DDIMScheduler
from diffusers.utils import export_to_video
import numpy as np
Expand Down
8 changes: 1 addition & 7 deletions examples/community/rerender_a_video.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,6 @@
# See the License for the specific language governing permissions and
# limitations under the License.

import sys
from dataclasses import dataclass
from typing import Any, Callable, Dict, List, Optional, Tuple, Union

Expand All @@ -21,6 +20,7 @@
import torch
import torch.nn.functional as F
import torchvision.transforms as T
from gmflow.gmflow import GMFlow
from transformers import CLIPImageProcessor, CLIPTextModel, CLIPTokenizer

from diffusers.image_processor import VaeImageProcessor
Expand All @@ -32,12 +32,6 @@
from diffusers.schedulers import KarrasDiffusionSchedulers
from diffusers.utils import BaseOutput, deprecate, logging
from diffusers.utils.torch_utils import is_compiled_module, randn_tensor


gmflow_dir = "/path/to/gmflow"
sys.path.insert(0, gmflow_dir)
from gmflow.gmflow import GMFlow # noqa: E402

from utils.utils import InputPadder # noqa: E402


Expand Down

0 comments on commit e64fdcf

Please sign in to comment.