Skip to content

How to run on cpu? #1

@waleedrazakhan92

Description

@waleedrazakhan92

I'm trying to run this on RTX quadro 3000 6gb and always get CUDA error as it overflows the memory. How do you i make it run only on 'cpu'. I've tried setting devices to 'cpu' in all of the config files but somehow there are still some tensors running on 'cuda' hence the mismatch error between devices.

  1. Could you also please provide code to run on cpu.
  2. What are the minimum VRAM requirements to run on gpu.
  3. Could you also provide a simpler way to load just the model as i want to edit using my own boundaries. So what i want to do is: 1) Load the model or cpu/gpu and load and image on cpu/gpu. 2) Use the model to get latent and save the latent. 3) Edit the latent 4) Provide the edited image using the generator.

I've been using StyleRes for few months now and have noticed that your model has superior inversion quality. The StyleRes runs fine on my gpu without any memory issues and that also uses E4E encoder. So i was wondering if i could just strip down your model and use it in place of StyleRes. But it seems quite convoluted and I haven't been able to do it.

Many Thanks for the work.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions